META
Software Engineer, Systems ML - Frameworks / Compilers / Kernels
META, New York, New York, us, 10261
Overview
Software Engineer, Systems ML - Frameworks / Compilers / Kernels. Join to apply for the
Software Engineer, Systems ML - Frameworks / Compilers / Kernels
role at
Meta . This role is part of the MTIA (Meta Training & Inference Accelerator) Software team within the PyTorch framework organization. The MTIA Software Team develops a comprehensive AI Compiler strategy to deliver a flexible platform to train and serve new DL/ML model architectures with auto-tuned high performance for production across specialized hardware. You will work on core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels, and tooling to accelerate ML workloads on MTIA hardware. You will collaborate with AI researchers to analyze models and lower them efficiently on MTIA hardware, and partner with hardware design teams to develop compiler optimizations for high performance. You will apply software development best practices to design features, optimization, and performance tuning, gaining experience in developing machine learning compiler frameworks and contributing to next-generation hardware-software co-design for AI problems. Responsibilities
Development of the software stack with core focus areas in AI frameworks, compiler stack, high-performance kernel development, and acceleration onto next-generation hardware architectures. Contribute to the PyTorch AI framework core compilers to support new AI hardware accelerators and optimize performance. Analyze deep learning networks and develop and implement compiler optimization algorithms. Collaborate with AI research scientists to accelerate next-generation models (e.g., recommendation systems, Generative AI, computer vision, NLP). Performance tuning and optimization of deep learning framework and software components. Minimum Qualifications
Proven C/C++ programming skills. Currently pursuing or having completed a Bachelor's degree in Computer Science, Computer Engineering, or a related technical field; degree must be completed prior to joining Meta. Experience in AI framework development or accelerating deep learning models on hardware architectures. Preferred Qualifications
Bachelor's degree with 4+ years of experience in AI framework development or accelerating deep learning models on hardware architectures, or a Masters degree with 2+ years, or a PhD in CS/CE or related fields. Knowledge of GPU, CPU, or AI hardware accelerator architectures. Experience with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT. Experience with AI kernel programming (CUDA, OpenMP/OpenCL) or accelerator kernel programming and accelerating libraries on AI hardware (e.g., cuBLAS, cuDNN, CUTLASS, HIP, ROCm). Experience with compiler optimizations (loop optimizations, vectorization, parallelization) and hardware-specific optimizations; familiarity with MLIR, LLVM, IREE, XLA, TVM, Halide is a plus. Experience developing training and inference framework components and system performance optimizations (latency, memory bandwidth, I/O, utilization) and related tooling. About Meta
Meta builds technologies that help people connect, find communities, and grow businesses. Meta is committed to Equal Employment Opportunity and Affirmative Action and does not discriminate based on race, religion, color, national origin, sex (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity or expression, transgender status, age, disability, or other legally protected characteristics. Meta participates in the E-Verify program where required by law. Meta may leverage artificial intelligence and machine learning in employment applications. Meta provides reasonable accommodations for candidates with disabilities; please request accommodations at accommodations-ext@fb.com. Compensation
Base pay range: $56.25/hour to $173,000/year, plus bonus, equity, and benefits. Individual compensation is determined by skills, qualifications, experience, and location. This posting reflects base compensation only. Seniority level
Not Applicable Employment type
Full-time Job function
Engineering and Information Technology Industries
Technology, Information and Internet
#J-18808-Ljbffr
Software Engineer, Systems ML - Frameworks / Compilers / Kernels. Join to apply for the
Software Engineer, Systems ML - Frameworks / Compilers / Kernels
role at
Meta . This role is part of the MTIA (Meta Training & Inference Accelerator) Software team within the PyTorch framework organization. The MTIA Software Team develops a comprehensive AI Compiler strategy to deliver a flexible platform to train and serve new DL/ML model architectures with auto-tuned high performance for production across specialized hardware. You will work on core areas such as PyTorch framework components, AI compiler and runtime, high-performance kernels, and tooling to accelerate ML workloads on MTIA hardware. You will collaborate with AI researchers to analyze models and lower them efficiently on MTIA hardware, and partner with hardware design teams to develop compiler optimizations for high performance. You will apply software development best practices to design features, optimization, and performance tuning, gaining experience in developing machine learning compiler frameworks and contributing to next-generation hardware-software co-design for AI problems. Responsibilities
Development of the software stack with core focus areas in AI frameworks, compiler stack, high-performance kernel development, and acceleration onto next-generation hardware architectures. Contribute to the PyTorch AI framework core compilers to support new AI hardware accelerators and optimize performance. Analyze deep learning networks and develop and implement compiler optimization algorithms. Collaborate with AI research scientists to accelerate next-generation models (e.g., recommendation systems, Generative AI, computer vision, NLP). Performance tuning and optimization of deep learning framework and software components. Minimum Qualifications
Proven C/C++ programming skills. Currently pursuing or having completed a Bachelor's degree in Computer Science, Computer Engineering, or a related technical field; degree must be completed prior to joining Meta. Experience in AI framework development or accelerating deep learning models on hardware architectures. Preferred Qualifications
Bachelor's degree with 4+ years of experience in AI framework development or accelerating deep learning models on hardware architectures, or a Masters degree with 2+ years, or a PhD in CS/CE or related fields. Knowledge of GPU, CPU, or AI hardware accelerator architectures. Experience with frameworks like PyTorch, Caffe2, TensorFlow, ONNX, TensorRT. Experience with AI kernel programming (CUDA, OpenMP/OpenCL) or accelerator kernel programming and accelerating libraries on AI hardware (e.g., cuBLAS, cuDNN, CUTLASS, HIP, ROCm). Experience with compiler optimizations (loop optimizations, vectorization, parallelization) and hardware-specific optimizations; familiarity with MLIR, LLVM, IREE, XLA, TVM, Halide is a plus. Experience developing training and inference framework components and system performance optimizations (latency, memory bandwidth, I/O, utilization) and related tooling. About Meta
Meta builds technologies that help people connect, find communities, and grow businesses. Meta is committed to Equal Employment Opportunity and Affirmative Action and does not discriminate based on race, religion, color, national origin, sex (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity or expression, transgender status, age, disability, or other legally protected characteristics. Meta participates in the E-Verify program where required by law. Meta may leverage artificial intelligence and machine learning in employment applications. Meta provides reasonable accommodations for candidates with disabilities; please request accommodations at accommodations-ext@fb.com. Compensation
Base pay range: $56.25/hour to $173,000/year, plus bonus, equity, and benefits. Individual compensation is determined by skills, qualifications, experience, and location. This posting reflects base compensation only. Seniority level
Not Applicable Employment type
Full-time Job function
Engineering and Information Technology Industries
Technology, Information and Internet
#J-18808-Ljbffr