Senior C++ Engineer – Machine Learning

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

• Design, develop, and optimize high-performance C++ code for ML compiler infrastructure
• Contribute to and maintain open-source projects related to compilers and ML systems
• Work with LLVM-based toolchains and compiler optimizations
• Optimize ML workloads for CPU and GPU architectures with a strong focus on performance
• Collaborate with cross-functional teams to integrate compiler solutions with ML frameworks
• Review code, propose architectural improvements, and mentor other engineers when needed

Requirements

• Master-level proficiency in C++ with strong knowledge of modern C++ standards
• Solid experience with LLVM and compiler internals
• Deep understanding of CPU performance, including memory hierarchy, vectorization, and parallelism
• Hands-on experience with open-source ML compilers targeting GPUs and/or CPUs
• Familiarity with TensorFlow and JAX ecosystems
• Proven experience working with GitHub and open-source development workflows

Nice to have

• Experience working in large-scale engineering environments (e.g., Google3 or similar monorepos)
• Background in ML model optimization, graph transformations, or kernel fusion
• Contributions to well-known open-source compiler or ML projects
• Experience with performance profiling and low-level debugging tools
• Knowledge of GPU architectures and optimization techniques

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.