Software Engineer, Systems ML - Frameworks / Compilers / Kernels | Ingénieur logiciel, Systèmes ML, cadres/Compilateurs/Noyaux
Meta1 day ago
Toronto, CanadaMid Level / Senior
Responsibilities
- Develop the software stack for AI frameworks, compiler systems, high-performance kernels, and acceleration on next-generation hardware architectures.
- Contribute to PyTorch core compiler development for training and inference AI accelerators and optimize their performance.
- Analyze deep learning networks and develop and implement compiler optimization algorithms.
- Collaborate with AI research scientists to accelerate deep learning models for recommendation systems, generative AI, computer vision, and NLP.
- Tune performance and optimize deep learning frameworks and software components.
Requirements
- Experience developing software in C/C++.
- Experience developing AI frameworks or accelerating deep learning models on hardware architectures.
- A bachelor's degree in computer science, computer engineering, or a relevant technical field, or equivalent practical experience.
- Preferred experience with compiler optimizations such as loop optimization, vectorization, parallelization, SIMD, MLIR, LLVM, IREE, XLA, TVM, or Halide.
- Preferred experience developing training and inference framework components and analyzing latency, memory bandwidth, I/O access, compute utilization, and system performance.
- Knowledge of GPU, CPU, or AI hardware accelerator architectures.
- Experience with PyTorch, Caffe2, TensorFlow, ONNX, or TensorRT.
- Preferred experience with CUDA, OpenMP, OpenCL, AI accelerator kernel programming, cuBLAS, cuDNN, CUTLASS, HIP, or ROCm.
- Experience with responsible and ethical AI practices and ongoing development of AI skills, including prompt/context engineering or agent orchestration.
- A bachelor's degree plus 7+ years, a master's degree plus 4+ years, or a PhD plus 3+ years of relevant AI framework or deep learning acceleration experience.
Benefits
- CA$133,000/year to CA$177,000/year plus bonus, equity, and benefits.
Categories
About Meta
Meta builds social platforms and communication apps—including Facebook, Instagram, WhatsApp, and Messenger—and develops AR/VR hardware and software such as Quest to power immersive computing. It monetizes primarily through advertising tools for businesses, with additional revenue from devices and services, and operates a massive global infrastructure. Founded in 2004 and headquartered in Menlo Park, California, Meta Platforms, Inc. is a public company traded on Nasdaq under the ticker META.