ML Systems Engineer - AI Trainer
Core
Designing GPU kernels, performance profiling, debugging, and inference serving for LLM systems and training infrastructure.
Role type
Senior IC ML Systems Engineer (GPU/LLM)
Builds
High-performance LLM training and inference systems
Domain
AI/ML Infrastructure, GPU Computing, LLMs
Deliverable
production ML models
Required skills
GPU kernel development, performance profiling, debugging, inference serving, MLOps, CUDA, Triton, JAX, PyTorch
Preferred skills
Familiarity with modern accelerators (A100, H100, TPU)
Technologies
CUDA, Triton, Kineto, torch.profiler, JAX, PyTorch, A100, H100, TPU
Responsibilities
Design challenging tasks involving GPU kernels, performance profiling, debugging, and inference serving; Develop guidelines and detailed rubrics for kernel optimization and profiler output interpretation; Evaluate MLOps and ML systems tasks and provide clear written technical feedback; Work with subject matter experts to maintain consistent, accurate training data.
