META
Software Engineer - Machine Learning Systems with PyTorch
META, Menlo Park, California, United States, 94029
Join Our Team!
We are seeking a talented Software Engineer to be part of the PyTorch Core Systems team, where you will play a crucial role in optimizing the software that powers AI models and systems. About Us:
The PyTorch team is dedicated to developing an open source software stack that serves as the backbone for state-of-the-art AI architectures. As a member of the Systems team, you will focus on enhancing performance for training and serving these advanced models. Your Responsibilities: Enhance PyTorch's capabilities for training, post-training, and inference on cutting-edge AI hardware accelerators. Develop and optimize PyTorch's software stack, concentrating on AI frameworks and high-performance kernel development. Engage in performance tuning and optimization for deep learning frameworks and software components. Collaborate with AI research scientists to expedite the development of next-generation deep learning models, including Recommendation Systems, Generative AI, Computer Vision, NLP, and more. Minimum Qualifications: Pursuing or in the process of completing a Bachelor's degree in Computer Science, Computer Engineering, or a related technical field (must be completed before joining). Proven skills in C/C++ programming. Experience in AI framework development or accelerating deep learning models on hardware architectures. Preferred Qualifications: Familiarity with GPU, CPU, or AI hardware accelerator architectures. Experience working with frameworks such as PyTorch, Caffe2, TensorFlow, ONNX, TensorRT, or high-performance AI kernels. Proficiency in CUDA programming, OpenMP/OpenCL programming, or programming for AI hardware accelerators. Knowledge in accelerating libraries similar to cuBLAS, cuDNN, CUTLASS, HIP, ROCm, etc. Experience in compiler optimizations including loop optimizations, vectorization, and hardware-specific optimizations like SIMD; familiarity with MLIR, LLVM, IREE, XLA, TVM, or Halide is a plus. Experience developing training and inference framework components as well as in system performance optimization. Compensation:
The public compensation range for this position is $56.25/hour to $173,000/year, plus bonus, equity, and benefits. Equal Opportunity:
Meta is proud to be an Equal Employment Opportunity and Affirmative Action employer. We embrace diversity and do not discriminate based on race, religion, color, national origin, sex, sexual orientation, gender identity, gender expression, disability, or other legally protected characteristics. We welcome applications from qualified candidates with criminal histories consistent with applicable laws. Accommodations:
Meta is committed to providing reasonable accommodations for candidates with disabilities in our recruiting process. If you need assistance or accommodations due to a disability, please reach out.
We are seeking a talented Software Engineer to be part of the PyTorch Core Systems team, where you will play a crucial role in optimizing the software that powers AI models and systems. About Us:
The PyTorch team is dedicated to developing an open source software stack that serves as the backbone for state-of-the-art AI architectures. As a member of the Systems team, you will focus on enhancing performance for training and serving these advanced models. Your Responsibilities: Enhance PyTorch's capabilities for training, post-training, and inference on cutting-edge AI hardware accelerators. Develop and optimize PyTorch's software stack, concentrating on AI frameworks and high-performance kernel development. Engage in performance tuning and optimization for deep learning frameworks and software components. Collaborate with AI research scientists to expedite the development of next-generation deep learning models, including Recommendation Systems, Generative AI, Computer Vision, NLP, and more. Minimum Qualifications: Pursuing or in the process of completing a Bachelor's degree in Computer Science, Computer Engineering, or a related technical field (must be completed before joining). Proven skills in C/C++ programming. Experience in AI framework development or accelerating deep learning models on hardware architectures. Preferred Qualifications: Familiarity with GPU, CPU, or AI hardware accelerator architectures. Experience working with frameworks such as PyTorch, Caffe2, TensorFlow, ONNX, TensorRT, or high-performance AI kernels. Proficiency in CUDA programming, OpenMP/OpenCL programming, or programming for AI hardware accelerators. Knowledge in accelerating libraries similar to cuBLAS, cuDNN, CUTLASS, HIP, ROCm, etc. Experience in compiler optimizations including loop optimizations, vectorization, and hardware-specific optimizations like SIMD; familiarity with MLIR, LLVM, IREE, XLA, TVM, or Halide is a plus. Experience developing training and inference framework components as well as in system performance optimization. Compensation:
The public compensation range for this position is $56.25/hour to $173,000/year, plus bonus, equity, and benefits. Equal Opportunity:
Meta is proud to be an Equal Employment Opportunity and Affirmative Action employer. We embrace diversity and do not discriminate based on race, religion, color, national origin, sex, sexual orientation, gender identity, gender expression, disability, or other legally protected characteristics. We welcome applications from qualified candidates with criminal histories consistent with applicable laws. Accommodations:
Meta is committed to providing reasonable accommodations for candidates with disabilities in our recruiting process. If you need assistance or accommodations due to a disability, please reach out.