About the Role

Develop AI compiler frameworks, PyTorch components, and high-performance kernels for Meta's MTIA AI hardware accelerators. Work on compiler optimizations, performance tuning, and collaborate with AI researchers to accelerate deep learning models. Focus on next-generation hardware-software codesign for AI workloads.

Key Responsibilities

• Development of SW stack with focus on AI frameworks, compiler stack, or high-performance kernel development
• Contribute to PyTorch AI framework core compilers for new AI hardware accelerators
• Analyze deep learning networks and implement compiler optimization algorithms
• Collaborate with AI research scientists to accelerate next-generation deep learning models
• Performance tuning and optimizations of deep learning framework & software components

Required Skills & Qualifications

Must Have:

• Proven C/C++ programming skills
• Experience in AI framework development or accelerating deep learning models on hardware architectures
• Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience

Nice to Have:

• Experience with compiler optimizations (loop optimizations, vectorization, parallelization) and tools like MLIR, LLVM, IREE, XLA, TVM, Halide
• Experience in developing training and inference framework components with system performance optimizations
• Experience with CUDA programming, OpenMP/OpenCL programming, or AI hardware accelerator kernel programming

Benefits & Perks

• bonus
• equity
• benefits