Research Scientist, Systems ML - Frameworks / Compilers / Kernels (PhD)

$117,000 - $173,000/Yr

Meta - Menlo Park, CA

Full-time - Senior

Menlo Park, CA

Web Search Portals, Libraries, Archives, and Other Information Services

About the position

The Research Scientist, Systems ML - Frameworks / Compilers / Kernels (PhD) position at Meta involves working with the Triton Accel team, contributing to the development of the PyTorch AI framework. The role focuses on research and development in AI frameworks, compiler stacks, and high-performance kernel development for AI accelerators. The position emphasizes open-source software while collaborating with internal product teams to optimize performance and support new AI hardware architectures.

Responsibilities

• Research and develop software stack focusing on AI frameworks, compiler stack, and high-performance kernel development.
• Contribute to the development of 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 enhance deep learning models in areas like recommendation systems, generative AI, and computer vision.
• Performance tuning and optimization of deep learning frameworks and software components.

Requirements

• Currently pursuing or obtained a Bachelor's degree in Computer Science, Computer Engineering, or a relevant technical field.
• Currently pursuing or obtained a PhD in Computer Science, Computer Engineering, or a relevant technical field.
• Proficiency in Python and C/C++ programming.
• Research experience in AI framework development or accelerating deep learning models on hardware architectures, demonstrated through peer-reviewed publications.
• Must obtain work authorization in the country of employment at the time of hire.

Nice-to-haves

• Knowledge of GPU, CPU, or AI hardware accelerator architectures.
• Experience with kernel-writing frameworks like Triton or CUDA.
• Experience with compiler optimizations such as loop optimizations, vectorization, and parallelization.
• Familiarity with MLIR, LLVM, IREE, XLA, TVM, Halide.

Benefits

• Competitive hourly wage of $56.25, with potential annual salary up to $173,000 plus bonuses and equity.
• Comprehensive benefits package including health insurance, retirement plans, and paid time off.