Senior Machine Learning Frameworks Engineer
Join our client’s innovative team dedicated to developing cutting-edge machine learning frameworks that streamline large-scale model training, fine-tuning, and evaluation. Their focus is on accelerating workflows and boosting model performance through continuous experimentation and infrastructure enhancements.
As a Senior ML Frameworks Engineer, you will play a key role in building and maintaining tools that support the entire machine learning lifecycle. Your work will enable efficient training, data processing, evaluation, and deployment of large language models (LLMs), empowering our teams and partners to achieve impactful results.
Essential functions
Develop robust machine learning tooling to facilitate model training, data ETL, evaluation, and deployment processes. Collaborate with both technical and non-technical stakeholders to create solutions tailored to specific use cases involving LLMs.
Deliver reusable, user-friendly tools that seamlessly integrate with existing data and ML systems. Contribute to continuous improvements in infrastructure and workflows, accelerating the pace of experimentation and deployment.
Qualifications
Strong understanding of machine learning principles, especially related to LLMs.
Solid professional experience in Python, with expertise in frameworks like Jax, TensorFlow, or PyTorch.
Skills in Java/Scala are preferred.
Proven experience in building scalable deep learning systems.
Experience working with large-scale data infrastructure.
Excellent verbal and written communication skills, capable of working effectively across diverse teams and organizations.
Would be a plus
Bachelor’s degree or higher in Computer Science or a related field
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 Senior Machine Learning Frameworks 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.