ML Runtime and Kernel Engineer - Core ML

Highlights

• Build runtime and kernels for novel LLM algorithms on wafer-scale engine.
• Turn research prototypes into high-performance training and inference demos.
• Optimize computation, memory, and communication for large-scale AI systems.

Ideal candidate

Systems-minded engineer who loves bridging ML research and efficient execution, thriving on profiling complex stacks and turning evolving ideas into robust, high-performance capabilities.

Requirements

• Bachelor's, Master's, PhD, or equivalent practical experience in Computer Science, Computer Engineering, Electrical Engineering, or related field.
• Experience developing high-performance systems software, ML systems, runtimes, compilers, or computational kernels.
• Strong programming skills in C++ and Python.
• Solid understanding of parallel programming, memory management, concurrency, data structures, and performance optimization.
• Proven ability to debug and profile complex software across multiple layers of a system.
• Familiarity with modern machine learning architectures and frameworks such as PyTorch or JAX.
• Ability to work effectively with researchers and translate evolving algorithmic requirements into reliable software.