joblaze
← Back to jobs

FSDP

5 jobs tagged FSDP

Kraken
Senior AI Compute Infrastructure Engineer Remote
Kraken United Kingdom Published 3 weeks ago
Ray TorchServe CUDA DeepSpeed vLLM
Apply →
Inflection AI
Member of Technical Staff – Model Training Remote Visa
Inflection AI Palo Alto, CA $175k–$350k/yr Published 11 months ago
RLAIF DeepSpeed DPO PyTorch GRPO
Apply →
Perplexity
Member of Technical Staff (Machine Learning Research Engineer)
Perplexity Berlin Published 1 month ago
DeepSpeed AI/ML PyTorch FSDP
Apply →
Databricks
Senior GenAI Research Engineer - Optimization and Kernels
Databricks Mountain View, California; San Francisco, California $166k–$225k/yr Published 6 months ago
NVIDIA CUDA DeepSpeed Megatron-LM PyTorch
Apply →
Together AI
Research Engineer, Frontier Speculative Decoding
Together AI San Francisco, New York City $190k–$270k/yr Published 6 months ago
DeepSpeed AI/ML Slurm PyTorch Python
Apply →