Expoint - all jobs in one place

המקום בו המומחים והחברות הטובות ביותר נפגשים

Limitless High-tech career opportunities - Expoint

Facebook Software Engineer Systems ML - PyTorch Compiler 
United States, California 
3178433

24.04.2025
This role is about developing the core PyTorch 2.0 technologies, innovating and advancing the state-of-the-art of ML compilers, and accelerating PT2 adoption through direct engagements with OSS and industry users.
Software Engineer, Systems ML - PyTorch Compiler, PyTorch Framework, PyTorch Performance Responsibilities
  • Develop the PT2 compiler (e.g., TorchDynamo, TorchInductor, Export, PyTorch Core).
  • Improve PyTorch performance via systematic solutions for the entire community.
  • Explore the intersection of the PyTorch compiler and PyTorch distributed.
  • Optimize Generative AI models across the stack (pre-training, fine-tuning, and inference).
  • Collaborate with users of PyTorch to enable new use cases of PT2 technologies both inside and outside Meta.
Minimum Qualifications
  • Advanced in Python or C++ programming.
  • Bachelor's degree in Computer Science, Computer Engineering, relevant technical field, or equivalent practical experience.
  • 4+ years of research or industry experience in developing compilers, ML systems, ML accelerators, GPU performance, and similar.
Preferred Qualifications
  • Experience in developing PyTorch/PT2, Triton, MLIR, JAX, XLA, TVM is a huge plus
  • Experience in developing inside other ML frameworks like Caffe2, TensorFlow, ONNX, TensorRT
  • Knowledge in GPU architecture, ML accelerator performance, and developing high-performance kernels.
  • Experience with training models, end-to-end model optimizations, or applying ML to systems.
  • Knowledge of communication collectives, PyTorch distributed, and parallelism.
  • Experience in building OSS communities and extensive social media presence in the ML Sys domain.
About Meta

$85.10/hour to $251,000/year + bonus + equity + benefits
Individual compensation is determined by skills, qualifications, experience, and location. Compensation details listed in this posting reflect the base hourly rate, monthly rate, or annual salary only, and do not include bonus, equity or sales incentives, if applicable. In addition to base compensation, Meta offers benefits. Learn more about at Meta.