Senior Software Engineer, ML Infrastructure
This position is part of a project for a world-renowned high-tech company, focused on building and improving the infrastructure that enables large scale machine learning frameworks used across the industry. Engineers on this project ensure that complex C++ and Python codebases, including core ML frameworks and supporting libraries, remain stable, efficient, and well integrated as they evolve. The role combines software development with build engineering and CI/CD automation, focusing on continuous enhancement of build and integration workflows and contributing to the ongoing evolution of these frameworks and their ecosystem.
Essential functions
Develop, refine, and enhance large scale build configurations for C++ and Python projects
Build and operate CI/CD pipelines to automate validation, testing, releases, and rollout of updates
Investigate and resolve complex build, dependency, and integration issues across multiple repositories
Implement code changes, fixes, and small features in ML frameworks and related libraries
Ensure reproducible and hermetic builds using modern toolchains, caching, and distributed testing
Manage and optimize containerized build and test environments (Docker)
Collaborate with infrastructure, release, and ML engineering teams to ensure consistent integration and delivery
Qualifications
Strong proficiency in C++ and Python
Experience with modern build systems at scale such as Bazel (preferred). Other experience with large build systems (Buck, Pants, CMake, or similar) is also valuable
Hands-on experience with CI/CD automation (GitHub Actions preferred, Jenkins, Buildkite, or similar)
Proficiency with Git or Mercurial including complex rebases, cherry-picks, and patch workflows
Strong Bash or shell scripting for automation and environment setup
Familiarity with Docker or similar container technologies for build and test automation
Detail-oriented, systematic approach to problem solving with focus on reliability and scalability
Bachelor’s/Master’s degree in Computer Science/ Engineering or a related field.
Would be a plus
Experience working with large open source ML frameworks such as TensorFlow, PyTorch, or JAX
Familiarity with GPU build and testing workflows or multi-architecture builds
Exposure to distributed or hermetic build environments and remote execution
Understanding of dependency graph analysis and build tooling such as Bazel query or cquery
We offer
- Opportunity to work on cutting-edge projects
- Work with a highly motivated and dedicated team
- Competitive salary
- Flexible schedule
- Benefits package - medical insurance, vision, dental, etc.
- 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
Information on personal data processing
You cannot apply for a position without accepting “INFORMATION ON PERSONAL DATA PROCESSING”
Applications for this job are no longer accepted. Please explore other open opportunities on our platform.
Thank you!
You applied for the position Senior Software Engineer, ML Infrastructure 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.