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

1X Technologies

1x.tech
1X Technologies is a humanoid robotics company founded in Norway in 2014 that develops general-purpose robots for everyday physical work. Its EVE wheeled android and the newer NEO home humanoid are trained with end-to-end neural networks rather than scripted motion, and the company runs its own hardware manufacturing. Backed by OpenAI and other investors, it operates engineering teams in Norway and the San Francisco Bay Area.
1x.tech • HQ: Moss, Norway • Robotics Hardware • Robotics AI • Robotics • Est. 2014
HQ: Moss, Norway • Robotics Hardware • Robotics AI • Robotics • Est. 2014
Top 25% payVerified live · 2 hours ago 3 months ago

AI Researcher

$250k – $350k per year • In office • Bachelor's Degree • Full-Time
Python
AI/ML
Fine-tuning
Multimodal AI
PyTorch
Synthetic Data
Tokenization
Pre-training
CUDA
CUDA Toolkit
DeepSpeed
Diffusion Models
Quantization
TensorRT
Triton
FSDP
Analytics
ETL/ELT
Apply
Report

Snowflake

snowflake.com
Snowflake is a cloud software company that helps organisations store, process, and analyse large volumes of data in a way that is designed for modern cloud infrastructure. Many businesses struggle with data spread across different systems, slow reporting cycles, and the cost and complexity of managing traditional data warehouses.
snowflake.com • HQ: Bozeman, United States • Database Management • Data & Analytics • Data Warehousing • 1001-5000 employees
HQ: Bozeman, United States • Database Management • Data & Analytics • Data Warehousing • 1001-5000 employees
Top 25% payVerified live · 3 hours ago 3 months ago

Intermediate/ Senior Software Engineer - Cortex LLM Training Platform

$200k – $288k per year • Senior • 6+ years expIn office (Bellevue, United States) • Bachelor's Degree • Full-Time • Senior
Snowflake
AI/ML
CUDA Toolkit
DeepSpeed
LLM
PyTorch
Ray
vLLM
CUDA
FSDP
NCCL
Post-training
SFT
AI Agents
DevOps
Cortex
Kubernetes
Prometheus
Apply
Report

Nuancelabs

nuancelabs.ai
Face-to-face AI interaction that feels human. We are building a human foundation model with emotional intelligence — it reads tone, expression, and even hesitation, and responds in real time.
nuancelabs.ai • Seattle • Software • Conversational AI • Artificial Intelligence • 51-200 employees • Est. 2025
Seattle • Software • Conversational AI • Artificial Intelligence • 51-200 employees • Est. 2025
Top 25% payVerified live · 3 hours ago 3 months ago

Member of Technical Staff — Pretraining Infra (Experienced)

$300k – $400k per year • Staff+ • In office (Seattle, United States) • Visa sponsorship • Staff
DeepSpeed
LLM
Multimodal AI
PyTorch
FSDP
Megatron-LM
NCCL
Synthetic Data
Post-training
DevOps
HPC
Apply
Report

Nokia

nokia.com
As a technology leader across mobile, fixed and cloud networks, our solutions enable a more productive, sustainable and inclusive world.
nokia.com • Singapore • Artificial Intelligence • Telecommunications • Cybersecurity • 1001-5000 employees • Est. 1865
Singapore • Artificial Intelligence • Telecommunications • Cybersecurity • 1001-5000 employees • Est. 1865
Verified live · 1 hour ago 3 months ago

Senior AI Engineer

Senior • In office • Bachelor's Degree • Senior
Python
Python
bandit
Mypy
AI/ML
AI Agents
AutoGen
CrewAI
Fine-tuning
Kubeflow
LangChain
LangGraph
Llama
LlamaIndex
LoRA
MLFlow
QLoRA
Qwen
RLHF
PEFT
SFT
DeepSpeed
KServe
RAG
Triton
vLLM
FSDP
TGI
DevOps
AWS
Azure
CI/CD
GCP
Helm
Kubernetes
Platform Engineering
Vector
Cybersecurity
SonarQube
Apply
Verified live · 1 hour ago 3 months ago

Senior AI Engineer

Senior • In office • Bachelor's Degree • Senior
Python
Python
bandit
Mypy
AI/ML
AI Agents
AutoGen
CrewAI
Fine-tuning
Kubeflow
LangChain
LangGraph
Llama
LlamaIndex
LoRA
MLFlow
QLoRA
Qwen
RLHF
PEFT
SFT
DeepSpeed
KServe
RAG
Triton
vLLM
FSDP
TGI
DevOps
AWS
Azure
CI/CD
GCP
Helm
Kubernetes
Platform Engineering
Vector
Cybersecurity
SonarQube
Apply
Report

Abbyy

abbyy.com
ABBYY Timeline is a leading process mining software with advanced process discovery & task mining capabilities for end-to-end business process visibility.
abbyy.com • HQ: Milpitas, United States • Cybersecurity • Artificial Intelligence • Data & Analytics • 501-1000 employees
HQ: Milpitas, United States • Cybersecurity • Artificial Intelligence • Data & Analytics • 501-1000 employees
Open 104 daysVerified live · 1 day ago 4 months ago

Senior Machine Learning Engineer, Model Training & Evaluation

$30k – $77k per year (Estimated) • Senior • 5+ years expRemote/Hybrid (Bengaluru, India) • Master's Degree • Senior
Python
AI/ML
Computer Vision
DeepSpeed
PyTorch
Quantization
FSDP
DevOps
Platform Engineering
Apply
Report

Saviynt

saviynt.com
We're building revolutionary identity and security solutions to help the world's largest companies migrate to the cloud and solve the toughest security challenges in record time. What's our secret? Our people. We're a global group of innovators wh...
saviynt.com • San Francisco • Bengaluru • Milpitas • Delhi • Atlanta • Artificial Intelligence • Cybersecurity • Identity Management • 501-1000 employees
San Francisco • Bengaluru • Milpitas • Delhi • Atlanta • Artificial Intelligence • Cybersecurity • Identity Management • 501-1000 employees
Top 25% payVerified live · 1 day ago 4 months ago

AI Platform Engineer, Training and Inference

$274k – $304k per year • Remote/Hybrid (Milpitas, United States) • Bachelor's Degree • Full-Time
Python
Databases
pgvector
Qdrant
PostgreSQL
AI/ML
Fine-tuning
LLM
MLFlow
ONNX
PyTorch
Quantization
RAG
Ray
Ray Serve
RLHF
RLlib
SGLang
TensorRT
vLLM
Triton
Flyte
FSDP
GRPO
NCCL
PPO
AWQ
Bitsandbytes
GPTQ
Post-training
DevOps
Google GKE
Vector
GCP
Kubernetes
Amazon S3
QA
k6
Apply
Report

Harmonic

harmonic.fm
Built for Artists. Curated by Fans. Create your own record store now.
harmonic.fm • Palo Alto • London • Publishing • Commerce • Marketplaces
Palo Alto • London • Publishing • Commerce • Marketplaces
Open 116 daysTop 25% pay 4 months ago

Research Engineer, Training & Inference

$200k – $450k per year • Senior • 5+ years expIn office (Palo Alto, United States) • Bachelor's Degree • Full-Time • Senior
Python
C++
C++
PyTorch C++
TensorFlow C++
AI/ML
CUDA
CUDA Toolkit
JAX
LLM
PyTorch
Reinforcement Learning
TensorFlow
NCCL
Triton
FSDP
Apply
Report

Blockchain Capital

blockchaincapital.com
Blockchain Capital is one of the earliest dedicated digital asset venture firms, founded in 2013 in San Francisco. It has invested across many funds in exchanges, infrastructure, decentralised finance and consumer applications. The firm was also a pioneer of tokenised fund structures, issuing a security token to represent limited partner interests.
blockchaincapital.com • HQ: San Francisco, United States • Blockchain & Crypto • Est. 2013
HQ: San Francisco, United States • Blockchain & Crypto • Est. 2013
Open 116 daysVerified live · 1 hour ago 4 months ago

Senior AI Compute Infrastructure Engineer

$90k – $170k per year (Estimated) • Senior • 5+ years expPanama • Remote (United Kingdom, Argentina, Brazil, Bulgaria, Canada) • Full-Time • Senior
Python
C++
Rust
AI/ML
KServe
Ray
Ray Serve
TensorRT
Triton
Triton Inference Server
vLLM
TorchServe
CUDA
CUDA Toolkit
DeepSpeed
AWS Trainium
FSDP
Megatron-LM
NCCL
TPU
DevOps
Kubernetes
AWS
Apply
Report

Profluent

profluent.bio
Profluent is a company founded in 2022 that trains large language models on protein sequences to design new functional proteins. It released OpenCRISPR-1, a gene editor designed entirely by a model and made freely available for ethical research and commercial use. The company works on editors, antibodies and enzymes that have no direct natural counterpart.
profluent.bio • HQ: Berkeley, United States • Biotechnology • Biopharma • Cell & Gene Therapy • Est. 2022
HQ: Berkeley, United States • Biotechnology • Biopharma • Cell & Gene Therapy • Est. 2022
Open 126 daysVerified live · 3 hours ago 5 months ago

Machine Learning Research Engineer

$132k – $248k per year (Estimated) • Middle • 3+ years expRemote/Hybrid (Emeryville, United States) • Bachelor's Degree • Middle
Python
AI/ML
CUDA
CUDA Toolkit
Fine-tuning
PyTorch
Spark
Triton
FSDP
SFT
DevOps
AWS
Azure
Docker
GCP
Kubernetes
Analytics
ETL/ELT
Apply
Report

Hedra

hedra.com
Create professional videos, images, and audio in one studio. Hedra gives you AI-powered tools to go from concept to production in minutes.
hedra.com • San Francisco • Content Creation • Artificial Intelligence • Generative Video
San Francisco • Content Creation • Artificial Intelligence • Generative Video
Open 139 daysTop 25% pay 5 months ago

Research Engineer

$175k – $275k per year • In office (San Francisco, United States) • Bachelor's Degree • Full-Time
DeepSpeed
Fine-tuning
Multimodal AI
PyTorch
Reinforcement Learning
FSDP
Post-training
Pre-training
PPO
Robotics
Reinforcement Learning
Sim-to-Real
Apply
Report

Andromeda

andromeda.ai
Andromeda connects AI teams with high-performance compute fast, at scale, and on terms that work. Buy, sell, and operate gpu clusters without complexity.
andromeda.ai • San Francisco • AI Infrastructure
San Francisco • AI Infrastructure
Open 144 daysVerified live · 2 hours ago 5 months ago

Senior Site Reliability Engineer

$81k – $170k per year (Estimated) • Senior • San Francisco • Remote (United States, Canada) • Full-Time • Senior
Bash
Python
AI/ML
CUDA Toolkit
DeepSpeed
PyTorch
CUDA
FSDP
InfiniBand
Megatron-LM
NCCL
NVLink
DevOps
Ansible
Grafana
Helm
Incident Management
Kubernetes
Prometheus
Self-Healing
SLURM
Terraform
HPC
Apply
Report

Dyna Robotics

dynarobotics.com
Dyna Robotics is a robotics foundation model company headquartered in Redwood City, California, and founded in 2024. The company trains general purpose manipulation models that let a single robot arm learn commercial tasks such as folding, packing, and handling from demonstration rather than explicit programming. It targets repetitive physical work in laundries, restaurants, and light manufacturing, and publishes model progress through its DYNA series of releases.
dynarobotics.com • HQ: Redwood City, United States • Multimodal AI • Service & Hospitality Robotics • Robotics AI • Artificial Intelligence • Robotic Software & Control Systems • Robotics • 51-200 employees • Est. 2024
HQ: Redwood City, United States • Multimodal AI • Service & Hospitality Robotics • Robotics AI • Artificial Intelligence • Robotic Software & Control Systems • Robotics • 51-200 employees • Est. 2024
Open 153 daysTop 25% pay 6 months ago

ML Infrastructure Engineer, Training

$220k – $320k per year • Senior • 7+ years expIn office (Redwood City, United States) • Full-Time • Senior
Accelerate
DeepSpeed
Knowledge Distillation
Multimodal AI
PyTorch
Quantization
TensorRT
Triton
FSDP
NCCL
DevOps
AWS
GCP
Kubernetes
SLURM
HPC
Apply
Report

Hark

hark.com
Hark was an online digital entertainment platform best known for its extensive library of short audio soundbites, video clips, and pop culture quotes. Launched in 2007, the website allowed users to browse, create, and share playable soundboards featuring memorable lines from movies, television shows, and political figures. While it grew into a popular destination for viral sound clips during the late 2000s and early 2010s, the platform has since ceased its original operations.
hark.com • HQ: San Jose, United States • Machine Learning • Artificial Intelligence
HQ: San Jose, United States • Machine Learning • Artificial Intelligence
Open 161 dayTop 25% pay 6 months ago

Member of Technical Staff, Multimodal Vision

$180k – $450k per year • Staff+ • In office (San Jose, United States) • Full-Time • Staff
DeepSpeed
Fine-tuning
Multimodal AI
Reinforcement Learning
RLHF
DPO
FSDP
GRPO
Megatron-LM
Post-training
PPO
SFT
AI Agents
Apply
Report

Build

build.io
Build is a cloud infrastructure and platform-as-a-service provider headquartered in London, United Kingdom, and founded in 2023. The company provides a full-stack platform for product teams to deploy and run production applications on its own bare-metal hardware rather than relying on rented hyperscaler capacity. It integrates AI-powered workflows for automated code deployment and infrastructure management, serving a global client base through data centers in the United States, Europe, and Japan.
build.io • HQ: London, United Kingdom • Web Development • Software • Data Centers • AI Agents • Artificial Intelligence • DevOps • Information Technology • Cloud Computing • Est. 2023
HQ: London, United Kingdom • Web Development • Software • Data Centers • AI Agents • Artificial Intelligence • DevOps • Information Technology • Cloud Computing • Est. 2023
Open 164 daysVerified live · 1 day ago 6 months ago

Member of Technical Staff - Post Training

$63k – $138k per year (Estimated) • Staff+ • Remote/Hybrid (Yokohama, Japan) • Full-Time • Staff
Python
AI/ML
DeepSpeed
LLM
PyTorch
Reinforcement Learning
Synthetic Data
vLLM
FSDP
Post-training
SFT
Apply
Report

Reflection AI

reflection.ai
Reflection AI is a company founded in 2024 by former Google DeepMind researchers who worked on AlphaGo and large language models. It builds autonomous coding agents and has committed to releasing frontier open-weight models as an American counterweight to Chinese open model releases. The company raised a very large round in 2025 to fund training at frontier scale.
reflection.ai • HQ: New York, United States • 501-1000 employees • Est. 2024
HQ: New York, United States • 501-1000 employees • Est. 2024
Open 172 daysVerified live · 1 day ago 6 months ago

Member of Technical Staff - Research Software Engineer

$184k – $345k per year (Estimated) • Equity • Staff+ • In office (New York, San Francisco, United States, London, United Kingdom) • Visa sponsorship • PhD • Full-Time • Staff
JAX
PyTorch
Ray
Reinforcement Learning
Tokenization
Triton
FSDP
Megatron-LM
NCCL
DevOps
Kubernetes
SLURM
Apply
Report

Cerebras Systems

cerebras.ai
Cerebras Systems is an American computer hardware company founded in 2016 and headquartered in Sunnyvale, California that builds accelerators for artificial intelligence at wafer scale. Instead of assembling clusters from many small chips, it manufactures a single processor the size of an entire silicon wafer, the Wafer Scale Engine, which removes most of the communication overhead in large model training and inference. The company sells CS-series systems to research laboratories and enterprises, operates its own inference cloud known for very high token throughput, and has built large supercomputers with partners including the Gulf technology group G42.
cerebras.ai • HQ: Sunnyvale, United States • Machine Learning • Artificial Intelligence • Hardware • Semiconductors • AI Infrastructure • 1001-5000 employees • Est. 2016
HQ: Sunnyvale, United States • Machine Learning • Artificial Intelligence • Hardware • Semiconductors • AI Infrastructure • 1001-5000 employees • Est. 2016
Verified live · 1 day ago 6 months ago

Applied Machine Learning Research Scientist

$231k – $433k per year (Estimated) • Middle • 4+ years expIn office • Bachelor's Degree • Full-Time • Middle
Python
AI/ML
Cerebras
Fine-tuning
LLM
PyTorch
Reinforcement Learning
RLHF
Synthetic Data
Edge AI
GRPO
OpenAI
Post-training
AI Agents
FSDP
Megatron-LM
Apply
Report

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 · 2 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 367 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
Firmus Technologies builds immersion-cooled artificial intelligence factories that run large GPU fleets on renewable power. Founded in 2021 in Singapore, it develops both the data centre design and the cloud service on top. Its Project Southgate campuses in Australia are among the region's largest planned artificial intelligence sites.
firmus.ai • HQ: Singapore • Artificial Intelligence • Est. 2021
HQ: Singapore • Artificial Intelligence • Est. 2021
Open 228 daysVerified live · 11 hours ago 8 months ago

AI Engineer, AI & Applications

$90k – $227k per year (Estimated) • Senior • 5+ years expIn office (Singapore) • Full-Time • Senior
DeepSpeed
Fine-tuning
JAX
Llama
LoRA
PyTorch
QLoRA
PEFT
FSDP
LLM Evaluation
Megatron-LM
NCCL
DevOps
Kubernetes
SLURM
Apply
Report

Liquid AI

liquid.ai
Liquid AI is an artificial intelligence company headquartered in Boston, Massachusetts, and founded in 2023 as a spin-off from the MIT Computer Science and Artificial Intelligence Laboratory. The company builds Liquid Foundation Models, an architecture derived from liquid neural networks that aims to match transformer quality at a fraction of the memory and compute. It targets on-device and edge deployment where models must run on phones, vehicles, and embedded hardware rather than in a data center.
liquid.ai • HQ: Boston, United States • AI Infrastructure • Hardware • Machine Learning • Edge AI • LLM & Generative AI • Artificial Intelligence • Est. 2023
HQ: Boston, United States • AI Infrastructure • Hardware • Machine Learning • Edge AI • LLM & Generative AI • Artificial Intelligence • Est. 2023
Open 258 daysVerified live · 1 day ago 9 months ago

Member of Technical Staff - Multi-Modal, Audio

$199k – $404k per year (Estimated) • Staff+ • Remote/Hybrid (San Francisco, Boston, United States) • Full-Time • Staff
DeepSpeed
Multimodal AI
PyTorch
FSDP
Text-to-Speech
Speech Recognition
Apply
Open 312 daysVerified live · 1 day ago 11 months ago

Member of Technical Staff - Multi-Modal, Vision

$194k – $394k per year (Estimated) • Staff+ • Remote/Hybrid (San Francisco, United States) • Master's Degree • Full-Time • Staff
Python
AI/ML
Computer Vision
DeepSpeed
Multimodal AI
Reinforcement Learning
VLM
FSDP
Hugging Face
Megatron-LM
Post-training
SFT
DevOps
GitHub
Apply
Open 398 daysVerified live · 1 day ago 1 year ago

Member of Technical Staff - Distributed Training Engineer

$199k – $404k per year (Estimated) • Staff+ • Remote/Hybrid (San Francisco, United States) • Full-Time • Staff
DeepSpeed
Multimodal AI
PyTorch
FSDP
Megatron-LM
NCCL
Mixture of Experts
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 271 day 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
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.