Senior ML Frameworks Engineer

We are seeking a talented Senior ML Frameworks Engineer to develop and enhance scalable machine learning tooling that supports the entire ML lifecycle-from data processing and model training to evaluation and deployment.

You will collaborate with cross-functional teams to create reusable, efficient solutions tailored for large-scale LLM projects.

Our client is a global leader renowned for pioneering innovations in computer software, personal computers, and mobile tablets, transforming the tech landscape. The team uses advanced machine learning frameworks designed to streamline large-scale model training, fine-tuning, and evaluation. They focus on accelerating training workflows and boosting model performance through continuous experimentation and infrastructure enhancements.

Join us to be at the forefront of innovation in AI and machine learning!

Essential functions

• Build machine learning tooling to facilitate various phases of the ML lifecycle from model training, data ETL, end-to-end model evaluation and deployment
• Work with technical and non-technical stakeholders to build solutions to align LLMs for specific use cases.
• Deliver reusable and easy-to-use tooling to integrate with existing data and machine learning systems

Qualifications

• Strong understanding of machine learning principles, especially in the context of LLMs.
• Solid proficiency in Python, including machine learning packages like Jax/Tensorflow or PyTorch
• Experience with large-scale data infrastructure
• Strong verbal and written communication skills with the ability to work effectively across internal and external organisations and virtual teams.

Would be a plus

• Skills in Java or Scala
• BS/BA or equivalent degree in computer science or similar

We offer

• Opportunity to work on bleeding-edge projects
• Work with a highly motivated and dedicated team
• Benefits package - medical insurance, sports
• Corporate social events
• 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.