This position is part of a project for a world-renowned high-tech company focused on the foundational infrastructure behind large scale AI workloads. The team works with core ML compiler and runtime systems that enable AI execution on CPUs, GPUs, and specialized accelerators such as TPUs. The project centers on adapting, stabilizing, and validating complex, performance critical systems so they can operate reliably across different environments and release contexts, while remaining aligned with the evolution of modern AI frameworks.
Essential functions
- 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
- Contribute to the ongoing stabilization and refinement of AI infrastructure used at scale
Qualifications
- 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
- Hands on experience working in containerized development or CI environments (Docker or similar)
- Working knowledge of Python for general engineering tasks and tooling
Would be a plus
- Exposure to large scale ML frameworks such as TensorFlow or JAX and their compilation or execution paths
- 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
- 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 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.