Senior Software Engineer, Kernels and Performance, Core ML Frameworks

Job description

Minimum qualifications: - Bachelor’s degree or equivalent practical experience. - 5 years of experience with software development in C++ or Python. - 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture. - Experience with performance optimization. Preferred qualifications: - Experience optimizing TPU/GPU code, using low-level kernel languages like Pallas, Compute Unified Device Architecture (CUDA), or Triton. - Knowledge of ML Frameworks (JAX/PyTorch), common operations like attention and Mixture of Experts (MoEs), including model optimization and low-precision formats. - Understanding of modern accelerators (e.g., data movement, pipelining, heterogeneous compute, and scale-out). - Understanding of compiler principles (optimization, code generation) and toolchains such as MLIR, OpenXLA. - Demonstrate a record of building developer infrastructure, including Open-Source Software (OSS) libraries, flexible high-performance APIs, and easy-to-consume documentation to empower the community. - Excellent investigative and problem-solving capabilities with communication skills across cross-functional teams....