C++ Machine Learning Engineer, Models Training

Job summary

Responsibilities

This offer of employment is contingent upon the applicant being eligible to access U.S. export-controlled technology. Due to U.S. export laws, including those codified in the U.S. Export Administration Regulations (EAR), the Company is required to ensure compliance with these laws when…

• How ML frameworks and compilers connect at the system level.
• How to translate training workloads into low-level operations optimized for custom silicon.
• How large-scale model training works under the hood, from memory layout to operator fusion.
• What it takes to build infrastructure that supports fast iteration in research and production.

What they're looking for

• Strong in C++ and low-level systems programming, especially in performance-critical code.
• Comfortable thinking in tensors, memory layout, and compiler graphs.
• Familiar with PyTorch and curious about how frameworks map to hardware.
• A builder who enjoys solving technical puzzles and digging into the details.

Tech stack

• C++
• PyTorch