Member of Technical Staff - Efficient ML

Moonlake AI(2 months ago)

San Mateo, CAOnsiteFull TimeSenior$175,492 - $236,344 (estimated)Engineering & Research

About this role

The Member of Technical Staff - Efficient ML at Moonlake focuses on optimizing machine learning training efficiency and inference processes with an emphasis on techniques such as gradient checkpointing, low-latency serving, and various optimization strategies including quantization and pruning. The role includes enhancing GPU and kernel performance through advanced profiling, as well as ensuring infrastructure reliability with multi-node job management and GPU failure handling. Collaboration in an on-site team environment is required to develop real-time interactive content solutions.

View Original Listing

Required Skills

Dataloaders
Fusion
Gradient Checkpointing
FSDP
ZeRO
Tensor Parallel
NCCL Tuning
Nsight Profiling
Triton
CUDA

+16 more

About Moonlake AI

moonlakeai.com

Moonlake AI is an innovative platform designed to empower users to create and code interactive games and virtual worlds seamlessly. The company focuses on delivering intuitive tools that allow users to bring their creative visions to life, making game development accessible to a broader audience. By integrating advanced AI technology into its services, Moonlake AI is positioned as a pioneer in reshaping how games and interactive experiences are conceived and built.