More jobs:
Member of Technical Staff – Efficient ML
Job in
San Mateo, San Mateo County, California, 94409, USA
Listed on 2026-03-01
Listing for:
Embedding VC
Full Time
position Listed on 2026-03-01
Job specializations:
-
IT/Tech
AI Engineer, Systems Engineer
Job Description & How to Apply Below
Introducing Moonlake, AI for creating world simulations.
Scope of WorkTraining efficiency
- Data loaders, fusion, activation remat, gradient checkpointing.
- FSDP/ZeRO/tensor+pipeline parallel; NCCL tuning.
GPU + kernel performance
- Nsight profiling, Triton/CUDA kernels, fused ops.
- Flash-attention–style speedups, sequence packing, KV-cache tricks.
Inference optimization
- Low-latency serving, continuous batching, speculative decoding.
- Quantization (GPTQ/AWQ), distillation, pruning.
Infra + reliability
- SLURM/K8s multi-node jobs, checkpoint hygiene.
- Determinism, env pinning, GPU failure handling.
We are committed to being an on-site, in-person team currently based in San Mateo.
#J-18808-LjbffrTo View & Apply for jobs on this site that accept applications from your location or country, tap the button below to make a Search.
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
(If this job is in fact in your jurisdiction, then you may be using a Proxy or VPN to access this site, and to progress further, you should change your connectivity to another mobile device or PC).
Search for further Jobs Here:
×