Skip to content
ModalModalNew York, NY

Member of Technical Staff - ML Performance

Engineers optimize ML systems for performance at scale, focusing on GPU utilization, inference engines, and container runtime to boost throughput and reduce latency for language and diffusion models. Requires 5+ years experience with PyTorch, CUDA, and performance debugging.

150k – 350k/yr
On-site5+ YOEML Engineering

About the role

Requirements

  • 5+ years of experience writing high-quality, high-performance code.
  • Experience working with torch, high-level ML frameworks, and inference engines (vLLM or TensorRT).
  • Familiarity with Nvidia GPU architecture and CUDA.
  • Experience with ML performance engineering (tell us a story about boosting GPU performance — debugging SM occupancy issues, rewriting an algorithm to be compute-bound, eliminating host overhead, etc).

Nice-to-have

  • Familiarity with low-level operating system foundations (Linux kernel, file systems, containers, etc).

Skills

PyTorchvLLMTensorRTCUDANvidia GpuMl FrameworksLinux KernelContainers

Similar roles

ML Engineering jobs
Sieve

Member of Technical Staff, Machine Learning

SieveSan Francisco, CA

Machine Learning Engineer owning the full ML lifecycle for multimodal video datasets at Sieve. Fine-tune VLMs, build evaluation/QA pipelines with frontier models, design filtering systems over internet-scale data, and ship production improvements for top AI labs. Requires strong Python, PyTorch, and production ML experience.

150k – 350k/yr
On-site5+ YOEML Engineering
Modal

Member of Technical Staff

ModalNew York, NY +1

Member of Technical Staff conducting hands-on LLM inference research at Modal. Own end-to-end bets on techniques like speculative decoding, quantization, KV-cache management, and disaggregation to improve cost per token and tail latency on production workloads. Requires strong LLM serving stack expertise and a track record shipping research or systems.

150k – 350k/yr
On-siteML Engineering
Parallel

Member of Technical Staff, Model Training

ParallelCalifornia

Own the training pipeline for search and agent models, building from product usage data through fine-tuning and evaluation to production deployment. Requires deep expertise in transformer fine-tuning, data curation, and training models for ranking, retrieval, and agent behavior.

150k – 300k/yr
On-siteML Engineering
Parallel

Member of Technical Staff, Search Ranking

ParallelUnited States

Own the multi-stage ranking pipeline for web-scale search, balancing precision, recall, latency, and compute cost across retrieval, first-pass ranking, and neural reranking.

150k – 300k/yr
On-site7+ YOEML Engineering
Second Nature

Staff Software Engineer, Engineering AI Team

Second NatureUnited States

Staff engineer builds AI-driven platform infrastructure for SDLC transformation, owns end-to-end experiments using AI agents like Claude, and ensures high-velocity code delivery with strong abstractions and real-world grounding. Requires staff-level architecture experience and AI-native workflows.

150k – 180k/yr
RemoteML Engineering