Posted 6 months ago
Member of Technical Staff - ML Infrastructure & Performance
AI Summary
A senior technical role focusing on ML infrastructure and performance optimization for real-time content generation, including GPU kernels, serving systems, and distributed training/processing. The role drives throughput, latency, and cost improvements.
About this role
Introducing Moonlake, AI for creating real-time interactive content
Mission: Improve Throughput, Latency, & Cost - deploying our models 2–10× faster & cheaper without quality regressions.
Scope of Work:
- GPU performance: CUDA/Triton kernels, FlashAttention family, paged attention, CUDA Graphs.
- Serving stack: TensorRT-LLM/Triton Inference Server, vLLM/TGI; continuous batching; on-GPU KV reuse; speculative decoding/medusa; mixture-of-agents routing.
- Parallelism: FSDP/ZeRO, TP/PP/expert parallel; NCCL tuning.
- Quantization/PEFT: AWQ/GPTQ/FP8; LoRA/DoRA serving.
- Systems: Ray/k8s/Argo, observability (Prom/Grafana/OpenTelemetry), autoscaling, A/B infra, canary + rollback.
Tech signals:
Previous experience at Infra-heavy startups such as Databricks, Roblox
We are committed to being an on-site, in-person team currently based in San Mateo
Skills
Explore related jobs
More jobs at Embedding VC
Similar A/b Testing jobs
CRO Manager (m/w/d) – Personalisierung & A/B-TestingJobs bei Lautsprecher Teufel GmbH · Berlin
Google Ads / SEA Marketing ManagerIn für Growth Hacking (m/w/d) | A/B-Testing | Performance MarketingNetzproduzenten® GmbH · Dresden, Germany
Vactor/Vacall Combination Truck DriverB&B Contracting Group · Surrey, Canada