FSDP Jobs - Remote & On-site

FSDP jobs hiring now at vetted startups and product companies on Alion, updated daily. Average salary around $298,009/year. Compare remote, hybrid and on-site positions with relocation and visa sponsorship. Browse and apply today.

AI/ML
Data Science
Backend
Frontend
Mobile
DevOps
Web3
Games
Hardware
Robotics
Security
QA
Executive
Networking
Product
Design
Analytics
Support
Enterprise Apps
Quantum

Baseten

baseten.co
Baseten is an AI infrastructure platform designed to help developers and machine learning teams deploy, serve, and scale open-source and custom AI models. The platform provides performant, low-latency inference infrastructure alongside developer tools like Truss, an open-source model packaging framework. Headquartered in San Francisco, California, Baseten enables companies to run state-of-the-art models in production seamlessly without managing underlying cloud infrastructure.
baseten.co • HQ: San Francisco, United States • Hardware • Machine Learning • AI Infrastructure • Artificial Intelligence
HQ: San Francisco, United States • Hardware • Machine Learning • AI Infrastructure • Artificial Intelligence
Top 25% payVerified live · 6 hours ago 8 months ago

Software Engineer - Training Product

$165k – $330k per year • Senior • 5+ years expRemote/Hybrid (San Francisco, New York, United States) • Full-Time • Senior
Cursor
DeepSeek
Reinforcement Learning
vLLM
PEFT
Axolotl
DeepSpeed
Fine-tuning
LoRA
PyTorch
Synthetic Data
FSDP
Megatron-LM
NCCL
Post-training
SFT
DevOps
Kubernetes
Apply
Open 366 daysTop 25% pay 1 year ago

Software Engineer - Training Infrastructure

$165k – $330k per year • Remote/Hybrid (San Francisco, New York, United States) • Bachelor's Degree • Full-Time
Go
Python
AI/ML
Cursor
Reinforcement Learning
Spark
DeepSpeed
PyTorch
FSDP
Megatron-LM
NCCL
DevOps
AWS
DigitalOcean
GCP
Kubernetes
Apply
Report

Mirelo AI

mirelo.ai
AI-powered sound and music generation for videos. We are passionate AI researchers, product experts and musicians.
mirelo.ai • Berlin • Tübingen • LLM & Generative AI • Corporate Events
Berlin • Tübingen • LLM & Generative AI • Corporate Events
Open 270 daysVerified live · 2 days ago 9 months ago

Research Scientist - Model Team

$87k – $207k per year (Estimated)Remote/Hybrid (Berlin, Tübingen, Germany) • PhD • Full-Time
Multimodal AI
Post-training
PyTorch
FSDP
Apply
Report

Cohere

cohere.com
Cohere is a Canadian AI company founded in 2019 and headquartered in Toronto. It builds secure, enterprise-focused large language models and AI tools for businesses and regulated industries. The company is known for emphasizing privacy, private deployment, and "sovereign AI" rather than consumer chat products
cohere.com • HQ: Toronto, Canada • Energy & Utilities • Biotechnology • Natural Language Processing • LLM & Generative AI • Artificial Intelligence • 1001-5000 employees • Est. 2019
HQ: Toronto, Canada • Energy & Utilities • Biotechnology • Natural Language Processing • LLM & Generative AI • Artificial Intelligence • 1001-5000 employees • Est. 2019
Verified live · 2 hours ago 9 months ago

Senior ML Systems Engineer, Frameworks & Tooling

$98k – $200k per year (Estimated) • Senior • LondonNew YorkParisSan FranciscoToronto • Remote (United States, France, United Kingdom, Canada) • Full-Time • Senior
Cohere SDK
CUDA Toolkit
JAX
LLM
Ray
CUDA
FSDP
NCCL
DeepSpeed
PyTorch
TensorRT
TensorRT-LLM
vLLM
xFormers
Megatron-LM
DevOps
Docker
Kubernetes
SLURM
HPC
Apply
Report

Basis

basis.ai
Basis is a nonprofit applied research organization with two mutually reinforcing goals. Our Mission Basis is a 501(c)(3) nonprofit applied research organization with two mutually reinforcing goals.
basis.ai • HQ: New York, United States • Biotechnology • Information Technology • Artificial Intelligence
HQ: New York, United States • Biotechnology • Information Technology • Artificial Intelligence
Verified live · 2 days ago 10 months ago

ML Systems Engineer, Infrastructure & Cloud

$144k – $315k per year (Estimated)In office (New York, Cambridge, United States) • Full-Time
JAX
PyTorch
Anthropic
FSDP
NCCL
OpenAI
DevOps
AWS
Azure
GCP
Kubernetes
Terraform
Apply
Report

Tahoe Therapeutics

tahoebio.ai
Tahoe Therapeutics generates very large single-cell perturbation datasets to train foundation models of how cells respond to drugs. Founded in 2023 in San Francisco, it released the Tahoe-100M dataset openly to the research community. Its models aim to predict drug effects across cancer cell contexts before laboratory testing.
tahoebio.ai • HQ: San Francisco, United States • Artificial Intelligence • Biotechnology • Health Care • Est. 2023
HQ: San Francisco, United States • Artificial Intelligence • Biotechnology • Health Care • Est. 2023
Open 286 daysTop 25% pay 10 months ago

Senior Machine Learning Scientist

$200k – $275k per year • Senior • In office (South San Francisco, United States) • PhD • Senior
JAX
Multimodal AI
NumPy
Pandas
PyTorch
SciPy
TensorFlow
Contrastive Learning
Flash Attention
Self-Supervised Learning
FSDP
Apply
Report

Entrata

entrata.com
Entrata provides an operating system for multifamily property management companies. Its modules cover leasing, accounting, resident payments, marketing websites and facilities. Large apartment owners use it as a single platform across their portfolios.
entrata.com • HQ: Lehi, United States • PropTech • Real Estate • Property Management • Est. 2003
HQ: Lehi, United States • PropTech • Real Estate • Property Management • Est. 2003
Open 402 daysTop 25% pay 1 year ago

Senior Machine Learning Engineer

$144k – $233k per year • Senior • 5+ years expRemote/Hybrid • Bachelor's Degree • Full-Time • Senior
Python
AI/ML
Fine-tuning
LLM
PyTorch
Synthetic Data
Post-training
SFT
AI Agents
DeepSpeed
vLLM
FSDP
Apply
Report
FSDP Jobs - Remote & On-site
Frequently asked questions
How many FSDP jobs are available on Alion?
There are currently 111 active listings from 70 hiring companies on Alion, refreshed daily.
What salary do FSDP jobs pay?
The average salary is around $298,009 per year based on the current listings.
Are remote options available for FSDP jobs?
Yes, many of these roles offer remote or hybrid work, and you can filter listings by remote, relocation and visa sponsorship.
How do I apply for FSDP jobs?
Open any listing on Alion and apply directly with the employer through the provided link - no lengthy sign-up required.