Lead/Principal Machine Learning Engineer
We are looking for a Lead/Principal Machine Learning Engineer with strong expertise in Java and AI/ML development to design and deliver scalable, high-performing B2B SaaS solutions. In this role, you will develop and deploy intelligent agents within multi-agent frameworks, build agentic workflows, and contribute to innovative AI-driven applications. You will also design and implement microservices using Java, Spring Boot, Kafka, and MySQL, ensuring secure, reliable, and efficient systems throughout the development lifecycle. This is a hands-on leadership role where you will influence architecture decisions, drive adoption of GenAI and LLM-based solutions, and deliver impactful, production-ready applications.
Essential functions
Develop agents in multi-agent frameworks and contribute to agentic workflow management and tool building.
Design, develop, test, and document microservices and enterprise applications using Java, Spring Boot, Kafka, and MySQL.
Apply appropriate standards, processes, and tools throughout the software development lifecycle to ensure secure, reliable solutions.
Participate in architecture design and reviews, and perform code reviews to verify quality and compliance with requirements.
Support code deployment across development, staging, and production environments, including authoring deployment procedures.
Troubleshoot and resolve production and security issues, ensuring system stability and performance.
Develop and optimize database queries, fix bugs, and improve the performance of existing applications.
Build and maintain large-scale, multi-tier applications, delivering highly scalable and high-performing solutions aligned with business goals.
Qualifications
Strong knowledge and working experience with GenAI, LLM, implementing and deploying AI Agents
6+ experience with Python in AI development
7+ experience with Core Java and Java web application development for RESTful web services
7+ years of strong knowledge with Spring Framework, Sprint boot, and Build Tools, such as Maven
5+ working knowledge in Reactive and Event-driven programming using Kafka
Strong OOP Design Skills and experience with design patterns
Ability to understand and author APIs
Practice in Test-driven Development
Good knowledge of multi-threading and concurrency
Would be a plus
Experience with CI/CD using Jenkins, Docker, Kubernetes
Experience in langchain, Langgraph, MVP, and other machine learning frameworks is a plus
We offer
- Opportunity to work on bleeding-edge projects
- Work with a highly motivated and dedicated team
- Competitive salary
- Flexible schedule
- Benefits package - medical insurance, sports
- Corporate social events
- Professional development opportunities
- Well-equipped office
About us
Grid Dynamics (NASDAQ: GDYN) is a leading provider of technology consulting, platform and product engineering, AI, and advanced analytics services. Fusing technical vision with business acumen, we solve the most pressing technical challenges and enable positive business outcomes for enterprise companies undergoing business transformation. A key differentiator for Grid Dynamics is our 8 years of experience and leadership in enterprise AI, supported by profound expertise and ongoing investment in data, analytics, cloud & DevOps, application modernization and customer experience. Founded in 2006, Grid Dynamics is headquartered in Silicon Valley with offices across the Americas, Europe, and India.Apply to the position
Thank you!
You applied for the position Lead/Principal Machine Learning Engineer successfully. We will get back to you soon. Have a great day!
Something went wrong...
There are possible difficulties with connection or other issues. Please try to use another browser (it's recommended to use the latest version of Google Chrome browser). If the problem still persists, please send your application to cv@griddynamics.com
RetrySomething went wrong...
Please double-check the information filled in the form, and make sure to provide valid data.
RetryDon’t see the right opportunity?
Contact us anyway and let’s talk! To apply, send your resume and cover letter to jobs@griddynamics.com
Grid Dynamics is an equal opportunity employer. We are committed to creating an inclusive environment for all employees during their employment and for all candidates during the application process.
All qualified applicants will receive consideration for employment without regard to, and will not be discriminated against based on, age, race, gender, color, religion, national origin, sexual orientation, gender identity, veteran status, disability or any other protected category. All employment is decided on the basis of qualifications, merit, and business need.