Sr. Software Engineer - ML Infrastructure

Grid Dynamics; Mexico; On-site; PHP 3,187,000 - 4,603,000; Full time

Benefits offered by this job

Health insurance

Vacation days

Life insurance

End-of-year bonus

Professional development

Job summary

Grid Dynamics seeks a Senior C++ Engineer to help build and optimize ML compiler infrastructure for CPUs and GPUs. You will tackle performance-critical systems, work with large C++ codebases, and collaborate with open-source communities across global teams.

You will contribute to portable, hermetic builds and scalable CI/CD pipelines while integrating Python tooling and multi-environment deployments. Based in Guadalajara, this role focuses on cutting-edge AI infrastructure and compiler-driven

Qualifications

• Strong proficiency in C++ with large codebases.
• Familiarity with LLVM or compiler systems.
• Understanding CPU architecture and performance considerations.
• Experience with open source ML compiler or runtime projects for CPU/GPU.
• Experience in Linux development environments.
• Python for tooling and general tasks.
• Experience with Bazel or large build systems.
• Hands-on CI/CD automation (GitHub Actions, Jenkins, Buildkite).
• Familiarity with Docker for build/test automation.

Responsibilities

• Work with large C++ codebases for ML compilation and runtime infrastructure.
• Adapt and evolve systems for multiple environments and deployment contexts.
• Improve portability, reliability, and consistency of core infrastructure components.
• Collaborate with compiler, framework, and infra teams for integration and validation.
• Develop, refine, and enhance large-scale build configurations for C++ and Python.
• Build and operate CI/CD pipelines to automate validation, testing, releases, and rollout of updates.
• Investigate and resolve build/dependency issues across repositories.
• Ensure reproducible and hermetic builds using modern tools and distributed testing.
• Manage and optimize containerized build/test environments (Docker).
• Contribute to stabilization and refinement of AI infrastructure used at scale.

Skills

C++ proficiency

LLVM familiarity

CPU architecture

Open source ML compiler

Linux development

Python tooling

Build systems (Bazel)

CI/CD automation

Docker

Job description

We are looking for a highly skilled Senior C++ Engineer to join our team working on cutting-edge machine learning compiler technologies for CPUs and GPUs. 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. In this role, you will contribute to performance-critical systems, collaborate with open-source communities, and help build scalable compiler infrastructure that powers modern ML frameworks.

Responsibilities

• Work with large existing C++ codebases related to ML compilation and runtime infrastructure
• Adapt and evolve existing systems to operate correctly across multiple environments and deployment contexts
• Improve portability, reliability, and consistency of core infrastructure components
• Collaborate with compiler, framework, and infrastructure teams to ensure correct integration and validation
• 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
• Ensure reproducible and hermetic builds using modern toolchains, caching, and distributed testing
• Manage and optimize containerized build and test environments (Docker)
• Contribute to the ongoing stabilization and refinement of AI infrastructure used at scale

Requirements

• Strong proficiency in C++ with experience working in large, mature codebases
• Familiarity with LLVM or compiler based systems
• Understanding of CPU architecture and performance considerations
• Familiarity with open source ML compiler or runtime projects targeting CPU or GPU
• Experience working in Linux based development environments
• Working knowledge of Python for general engineering tasks and tooling
• 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)
• Familiarity with Docker or similar container technologies for build and test automation

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
• Work around ML compiler stacks including XLA, MLIR, or LLVM based backends
• Interest in low level, performance sensitive AI infrastructure and compiler driven execution models

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
• Life Insurance and funeral expenses
• 5% savings fund, uncapped (matched by the company in the end of the year)
• 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)
• 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.