The Rundown AI, Inc.
Software Engineer - Systems ML - PyTorch
The Rundown AI, Inc., Menlo Park, California, United States, 94029
In this role, you will be a member of the PyTorch Core Systems team. The PyTorch team develops the open source software stack powering AI models and systems. The Systems team optimizes highly performant software to train and serve AI architectures. You will work closely with AI researchers to analyze deep learning models and optimize their performance within PyTorch. You will also partner with researchers to understand modern advances in AI guided software development and apply this directly towards PyTorch code and device optimization.
Software Engineer - Systems ML - PyTorch Responsibilities:
Improve PyTorch's training, post-training, and inference on modern AI hardware accelerators.
Develop PyTorch's software stack focusing on AI frameworks and high-performance kernel development.
Perform performance tuning and optimization of deep learning framework components.
Collaborate with AI research scientists to accelerate the next generation of deep learning models such as recommendation systems, generative AI, computer vision, NLP, etc.
Minimum Qualifications:
Currently pursuing or have completed a Bachelor's degree in Computer Science, Computer Engineering, or a relevant technical field prior to joining Meta. Proven C/C++ programming skills. Experience in AI framework development or accelerating deep learning models on hardware architectures.
Preferred Qualifications:
Knowledge of GPU, CPU, or AI hardware accelerator architectures. Experience with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT. Experience with CUDA, OpenMP/OpenCL programming, or AI hardware kernel programming, and accelerating libraries such as cuBLAS, cuDNN, CUTLASS, HIP, ROCm. Experience with compiler optimizations, MLIR, LLVM, IREE, XLA, TVM, Halide. 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. From Facebook to Messenger, Instagram, and WhatsApp, Meta empowers billions worldwide. Moving beyond 2D screens, Meta is exploring immersive experiences like AR and VR to shape the future of social technology. Join us to help build a future beyond today's digital connection limits. Meta is an Equal Employment Opportunity and Affirmative Action employer. We do not discriminate based on race, religion, color, national origin, sex, sexual orientation, gender identity, age, disability, or other protected characteristics. We also consider qualified applicants with criminal histories, in accordance with applicable law. Meta participates in the E-Verify program where required. Meta is committed to providing accommodations for candidates with disabilities. Contact us at accommodations-ext@fb.com for assistance. Compensation ranges from $56.25/hour to $173,000/year plus bonus, equity, and benefits. Compensation varies based on skills, qualifications, experience, and location. Benefits details are available at Meta. #J-18808-Ljbffr
Minimum Qualifications:
Currently pursuing or have completed a Bachelor's degree in Computer Science, Computer Engineering, or a relevant technical field prior to joining Meta. Proven C/C++ programming skills. Experience in AI framework development or accelerating deep learning models on hardware architectures.
Preferred Qualifications:
Knowledge of GPU, CPU, or AI hardware accelerator architectures. Experience with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT. Experience with CUDA, OpenMP/OpenCL programming, or AI hardware kernel programming, and accelerating libraries such as cuBLAS, cuDNN, CUTLASS, HIP, ROCm. Experience with compiler optimizations, MLIR, LLVM, IREE, XLA, TVM, Halide. 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. From Facebook to Messenger, Instagram, and WhatsApp, Meta empowers billions worldwide. Moving beyond 2D screens, Meta is exploring immersive experiences like AR and VR to shape the future of social technology. Join us to help build a future beyond today's digital connection limits. Meta is an Equal Employment Opportunity and Affirmative Action employer. We do not discriminate based on race, religion, color, national origin, sex, sexual orientation, gender identity, age, disability, or other protected characteristics. We also consider qualified applicants with criminal histories, in accordance with applicable law. Meta participates in the E-Verify program where required. Meta is committed to providing accommodations for candidates with disabilities. Contact us at accommodations-ext@fb.com for assistance. Compensation ranges from $56.25/hour to $173,000/year plus bonus, equity, and benefits. Compensation varies based on skills, qualifications, experience, and location. Benefits details are available at Meta. #J-18808-Ljbffr