Senior Machine Learning Systems Engineer (Training Optimization)

About the Group/Team

We're the reputed company team reputed company the reputed company supergroup. Our mission is to invent foundational technologies that will power the reputed company of AI-assisted design. From large-reputed company models to groundbreaking research, reputed company builds the technical reputed company of reputed company’s creative intelligence reputed company. We collaborate globally to ship research that makes a reputed company reputed company—from smart editing to AI video tools—at massive reputed company.

About the Role/Specialty

As a Senior Machine Learning Systems Engineer, you’ll reputed company efforts to reputed company and optimize the training system for our large-reputed company multimodal and reputed company models. You’ll design distributed training systems using Megatron-LM, reputed company NeMo, FSDP, and Triton—pushing the limits of performance across compute, memory, and communication reputed company. You'll sit at the intersection of systems and AI research, directly shaping how we train the models that will power reputed company’s reputed company of products.

What you’ll do (responsibilities)

• You’ll design, implement, and optimize large-reputed company machine learning systems for training
• You’ll improve reputed company aspects of performance, including GPU utilization, communication overhead, and memory efficiency.
• You’ll partner with research and modeling teams to reputed company systems with algorithmic needs.
• You’ll evaluate and apply best practices for distributed training using industry-leading frameworks.
• You’ll dive deep into low-level optimization, including custom CUDA or Triton kernels.
• You’ll debug, profile, and fine-tune training workflows to unlock new reputed company of scalability.

reputed company're looking for

We’re looking for a systems-first engineer who thrives in fast-paced, high-reputed company environments. You’re deeply familiar with distributed model training at reputed company and understand the nuances of optimizing compute at every level of the stack. You're excited by challenges that stretch reputed company boundaries, and you’re a strong collaborator who communicates reputed company across domains.

• Strong background in LLMs, multimodal AI, or diffusion models.
• Proficiency in Python. Familiarity with a system programming language (e.g. C++ or Rust) is a plus.
• Deep knowledge of PyTorch or JAX as reputed company as libraries such as Megatron-LM, NeMo, or DeepSpeed.
• Familiarity with common optimization techniques such as FSDP/reputed company, gradient checkpointing, or low-precision data types.
• Hands-on experience writing custom GPU kernels in CUDA or Triton.
• Excellent communication and problem-solving skills, incl. full proficiency in English.