**Job Title:** Software Engineer, Systems ML - Frameworks / Compilers / Kernels

**About the Role:**
In this role, you will be a member of the MTIA (Meta Training & Inference Accelerator) Software team, which is part of the industry-leading PyTorch AI framework organization. The MTIA Software Team is developing a comprehensive AI Compiler strategy that provides a flexible platform to train and serve new DL/ML model architectures, delivering auto-tuned high performance for production environments across specialized hardware architectures.

You will work on core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels, and tooling to accelerate machine learning workloads on current and next-generation MTIA AI hardware platforms. Collaboration with AI researchers will be essential as you analyze deep learning models and optimize them for MTIA hardware. You will also partner with hardware design teams to develop compiler optimizations for high performance.

**Responsibilities:**
- Develop software stack focusing on AI frameworks, compiler stack, and high-performance kernel development.
- Contribute to the development of PyTorch AI framework core compilers to support state-of-the-art inference and training AI hardware accelerators.
- Analyze deep learning networks and implement compiler optimization algorithms.
- Collaborate with AI research scientists to advance deep learning models in areas such as recommendation systems, generative AI, computer vision, and NLP.
- Optimize performance of deep learning frameworks and software components.

**Minimum Qualifications:**
- 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, or a relevant technical field, or equivalent practical experience.

**Preferred Qualifications:**
- A Bachelor’s degree in Computer Science, Computer Engineering or a relevant field with 7+ years of experience OR a Master's degree with 4+ years of experience OR a PhD with 3+ years of experience in AI framework development or accelerating deep learning models on hardware architectures.
- Knowledge of GPU, CPU, or AI hardware accelerator architectures.
- Experience working with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT.
- Experience with CUDA, OpenMP/OpenCL programming or AI hardware accelerator kernel programming.
- Knowledge of compiler optimizations and relevant projects such as MLIR, LLVM, IREE, XLA, TVM, Halide is a plus.
- Experience in developing training and inference framework components and system performance optimizations.

**About Meta:**
Meta builds technologies that help people connect, find communities, and grow businesses. Meta is advancing beyond 2D screens toward immersive experiences, such as augmented and virtual reality, to redefine social technology.

**Compensation:**
CA$133,000/year to CA$176,000/year + bonus + equity + benefits. Individual compensation is determined by skills, qualifications, experience, and location. In addition to base compensation, Meta offers benefits.

For more information about benefits at Meta, please visit our website.