Sr. 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.

Responsibilities

• 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

Requirements

• 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

Nice to Have

• 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:

• 100% payroll scheme, benefits by law (IMSS, INFONAVIT, 12+ vacation days)
• Benefits above the law: Vacation premium 50%, 5 PTOs, 3 sick days, 10 guaranteed public holidays per year
• Major medical insurance, Dental and Vision plan for an employee and direct family members
• Minor Medical Insurance (Multiservicios Médicos Santander) for an employee and direct family members
• Life Insurance and funeral expenses
• 5% savings fund, uncapped (matched by the company in the end of the year)
• Grocery cards/vouchers (Vales de Despensa)
• 30 days End of the Year Bonus (Aguinaldo)
• Opportunity to work on bleeding-edge projects with a highly motivated and dedicated team all over the world
• Individual career development plan and support from the best experts
• Professional development opportunities (Linkedin Learning, Cloud certification programs, access to corporate LMS integrated with other learning platforms)
• Well-equipped office in a business area of Guadalajara (quiet room, games room, air hockey, PS5, Nintendo Switch and Xbox Series X, pool table, ping pong, snacks, smoothies, and much more)
• Corporate social events (yoga, massages, sport tournaments, discussion panels, technical talks, lunch & learns)
• Flexible working hours
• Opportunity to relocate to another country where the company's offices are present.

About Us:

Grid Dynamics (Nasdaq:GDYN) is a digital-native technology services provider that accelerates growth and bolsters competitive advantage for Fortune 1000 companies. Grid Dynamics provides digital transformation consulting and implementation services in omnichannel customer experience, big data analytics, search, artificial intelligence, cloud migration, and application modernization. Grid Dynamics achieves high speed-to-market, quality, and efficiency by using technology accelerators, an agile delivery culture, and its pool of global engineering talent. Founded in 2006, Grid Dynamics is headquartered in Silicon Valley with offices across the US, UK, Netherlands, Mexico, and Central and Eastern Europe.

To learn more about Grid Dynamics, please visit www.griddynamics.com. Follow us on Facebook, Twitter, and LinkedIn.