Let’s get started
By clicking ‘Next’, I agree to the Terms of Service
and Privacy Policy, and consent to receive emails from Rise
Jobs / Job page
Member of Technical Staff - Efficient ML image - Rise Careers
Job details

Member of Technical Staff - Efficient ML

Introducing Moonlake, AI for creating world simulations.

Scope of Work

Training efficiency

  • Dataloaders, 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

Average salary estimate

$240000 / YEARLY (est.)
min
max
$180000K
$300000K

If an employer mentions a salary or salary range on their job, we display it as an "Employer Estimate". If a job has no salary data, Rise displays an estimate if available.

Similar Jobs
Photo of the Rise User
Posted 23 hours ago

Lead development and validation of gene delivery strategies (viral and non-viral) to enable expression of next-generation molecular neural interfaces at Merge Labs' San Francisco bio lab.

Photo of the Rise User
Helm.ai Hybrid No location specified
Posted 2 hours ago

Helm.ai seeks a Research Engineer to research and implement state-of-the-art model optimization techniques and deploy optimized deep learning models on GPUs and AI accelerators.

bdx Hybrid USA NJ - Franklin Lakes
Posted 11 hours ago

Lead technical R&D efforts developing test methods, fixtures, and manufacturable designs for medical device/IVD specimen management products in a high‑volume regulated environment.

Photo of the Rise User
Helm.ai Hybrid No location specified
Posted 16 hours ago

Work on state-of-the-art unsupervised learning and perception research at Helm.ai, turning rapid experimental results into deployable algorithms for autonomous vehicles.

Photo of the Rise User

Toyota Research Institute is hiring a Senior Robotics Research Engineer to design and integrate hardware-software systems that advance general-purpose robotic manipulation in real-world environments.

MATCH
Calculating your matching score...
FUNDING
SENIORITY LEVEL REQUIREMENT
TEAM SIZE
No info
HQ LOCATION
No info
EMPLOYMENT TYPE
Full-time, onsite
DATE POSTED
January 17, 2026
Risa star 🔮 Hi, I'm Risa! Your AI
Career Copilot
Want to see a list of jobs tailored to
you, just ask me below!