Member of Technical Staff - Efficient ML

Embedding Vc

San Francisco, California, United States permanent

Posted: January 15, 2026

Interested in this position?

Create a free account to apply with AI-powered matching

Quick Summary

We are looking for a skilled engineer to join our team as a member of our technical staff, where you will be working on efficient machine learning projects, including training efficiency, inference optimization, and infrastructure development.

Required Skills

Machine Learning Dataloaders Gradient Checkpointing Distributed Training FSDP ZeRO Tensor Parallelism NCCL Tuning GPU Performance Nsight Profiling CUDA Kernels Flash-Attention

Job Description

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

Why Apply Through MisuJob?

AI-Powered Job Matching: MisuJob uses advanced artificial intelligence to analyze your skills, experience, and career goals. Our matching algorithm compares your profile against thousands of job requirements to find positions where you have the highest chance of success. This saves you hours of manual job searching and ensures you only see relevant opportunities.

One-Click Applications: Once you create your profile, applying to jobs is effortless. Your resume and cover letter are automatically tailored to highlight the most relevant experience for each position. You can apply to multiple jobs in minutes, not hours.

Career Intelligence: Beyond job matching, MisuJob provides valuable career insights. See how your skills compare to market demands, identify skill gaps to address, and understand salary benchmarks for your experience level. Make data-driven decisions about your career path.

Frequently Asked Questions

How do I apply for this position?

Click the "Register to Apply" button above to create a free MisuJob account. Once registered, you can apply with one click and track your application status in your dashboard.

Is MisuJob free for job seekers?

Yes, MisuJob is completely free for job seekers. Create your profile, get matched with jobs, and apply without any cost. We help you find your dream job without any hidden fees.

How does AI matching work?

Our AI analyzes your resume, skills, and experience to understand your professional profile. It then compares this against job requirements using natural language processing to calculate a match percentage. Higher matches mean better fit for the role.

Can I apply to jobs in other countries?

Absolutely. MisuJob features jobs from companies worldwide, including remote positions. Filter by location or look for remote opportunities to find jobs that match your preferences.

Ready to Apply?

Join thousands of job seekers using MisuJob's AI to find and apply to their dream jobs automatically.

Create Free Account Sign In

← Browse all jobs