Software Engineer, Systems ML - Frameworks / Compilers / DL-Kernels | Ingénieur logiciel spécialisé en apprentissage automatique des systèmes – cadres/compilateurs/noyaux d'apprentissage profond
About the role
Role within MTIA Software team, part of AI and Compute Foundations. Focuses on PyTorch/GenAI framework components, AI compiler and runtime, high-performance DL kernels, and tooling to accelerate ML workloads on MTIA hardware. Collaborates with AI researchers and hardware design teams to optimize models for MTIA platforms; emphasizes software engineering practices and next-generation hardware-software co-design for AI.
À ce poste, vous ferez partie de l'équipe logicielle MTIA et intégrerez l'équipe plus large chargée de l'IA et des fondements informatiques, en travaillant sur des cadres/frameworks, compilateurs, noyaux d'apprentissage profond et outils pour MTIA.
Responsibilities
- Development of SW stack with core focus areas: AI frameworks, compiler stack, high performance deep learning kernel development, runtime and tooling for MTIA accelerator
- Contribute to PyTorch AI framework, core compilers, and GenAI OSS frameworks to support state-of-the-art AI hardware accelerators and optimize performance
- Analyze deep learning models, develop and implement optimization algorithms for high performance and efficiency gains
- Collaborate with AI research scientists to accelerate next generation models such as recommendation systems and Generative AI
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 related technical field, or equivalent practical experience
- Preferred: Experience with PyTorch, TensorFlow, ONNX, TensorRT, GenAI frameworks (e.g., vLLM, SGLang)
- Preferred: Experience with AI compilers (MLIR/LLVM/IREE/XLA/TVM/Halide), loop optimizations, vectorization, parallelization, SIMD
- Preferred: Runtime development and system performance optimizations
- Preferred: High performance DL kernels programming (CUDA, OpenMP/OpenCL) or acceleration libraries on AI hardware (cuBLAS/cuDNN/CUTLASS/HIP/ROCm)
- Preferred: Tooling for device profiling, tracing, debugging, performance tuning
- Knowledge of GPU/CPU/AI hardware accelerator architectures
- Demonstrated ongoing AI skill development and ability to integrate AI tools to optimize workflows
- Commitment to responsible, ethical AI practices
Skills
Benefits
- Bonus
- Equity
- Benefits
About Meta
Shop the latest AI glasses and wearable technology from Meta.
Shop the latest AI glasses and wearable technology from Meta. Explore Ray-Ban Meta, Oakley Meta, display glasses, and cutting-edge VR headsets.
More open roles at Meta
- Research Scientist Intern, 3D-grounded Generalist Model for Robotics (PhD)3d agoRedmond, WA · Artificial Intelligence
- 3d ago
- Research Scientist Intern, Robotics - Meta Reality Labs Research (PhD)3d agoZurich, Switzerland · Artificial Intelligence
- 3d ago
- Product Designer, Instagram3d agoHybrid (New York, NY) · Design