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

Tzafon

tzafon.ai
Tzafon is advancing machine intelligence. Our team develops models and systems for autonomous agents interacting with the real world.
tzafon.ai • San Francisco • AI Agents
San Francisco • AI Agents
Open 325 daysTop 25% pay 10 months ago

Member of Technical Staff - Foundations

$200k – $500k per year • Equity • Staff+ • Remote/Hybrid (San Francisco, United States) • Visa sponsorship • Full-Time • Staff
Python
AI/ML
CUDA
CUDA Toolkit
Fine-tuning
JAX
PyTorch
Triton
FSDP
Post-training
Pre-training
TPU
Synthetic Data
vLLM
Anthropic
OpenAI
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 311 daysVerified live · 1 day ago 10 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 257 daysVerified live · 1 day ago 8 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 397 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

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 163 days 5 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

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 160 daysTop 25% pay 5 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

PhysicsX

physicsx.ai
PhysicsX is a deep technology company that develops artificial intelligence and simulation software to accelerate engineering and advanced manufacturing. Founded by former Formula One engineers and computer scientists, the organization uses physics-informed AI and deep learning to optimize complex designs in fields like aerospace, automotive, and renewable energy. By replacing traditional, time-intensive physics simulations with real-time machine learning models, it enables companies to innovate faster and dramatically improve performance.
physicsx.ai • HQ: London, United Kingdom • Industrial Engineering • Artificial Intelligence • Simulation & Digital Twin
HQ: London, United Kingdom • Industrial Engineering • Artificial Intelligence • Simulation & Digital Twin
Verified live · 5 hours ago 25 days ago

Senior Machine Learning Infrastructure Engineer, Research

$96k – $241k per year (Estimated) • Equity • Senior • 5+ years expRemote/Hybrid (Singapore) • Bachelor's Degree • Senior
Python
AI/ML
Fine-tuning
MLFlow
PyTorch
Transformers
CoreWeave
FSDP
InfiniBand
NCCL
NVLink
DevOps
CI/CD
Grafana
Kubernetes
Octopus Deploy
Prometheus
SLURM
HPC
Apply
Verified live · 5 hours ago 4 months ago

Principal Machine Learning Infrastructure Engineer

$110k – $232k per year (Estimated) • Equity • Lead • 5+ years expRemote/Hybrid (London, United Kingdom) • Bachelor's Degree • Principal
Python
AI/ML
Fine-tuning
MLFlow
PyTorch
Transformers
CoreWeave
FSDP
InfiniBand
NCCL
NVLink
DevOps
CI/CD
Grafana
Kubernetes
Octopus Deploy
Prometheus
SLURM
HPC
Apply
Report

Adobe

adobe.com
Adobe is an American software company founded in 1982 by John Warnock and Charles Geschke and headquartered in San Jose, California. It created the PostScript and PDF formats and built the industry-standard creative toolset around Photoshop, Illustrator, Premiere Pro, After Effects and InDesign, now delivered as the Creative Cloud subscription. The company also runs Document Cloud for electronic signatures and document workflows, Experience Cloud for enterprise marketing analytics and personalisation, and the commercially safe Firefly family of generative AI models.
adobe.com • HQ: San Jose, United States • Sales & Marketing • Document Management • Artificial Intelligence • LLM & Generative AI • Software • Design & Creative • Graphic Design • Productivity Software • Digital Design • 5000+ employees • Est. 1982
HQ: San Jose, United States • Sales & Marketing • Document Management • Artificial Intelligence • LLM & Generative AI • Software • Design & Creative • Graphic Design • Productivity Software • Digital Design • 5000+ employees • Est. 1982
Top 25% payVerified live · 1 day ago 6 days ago

Senior Applied Scientist / Engineer, Training & Inference

$164k – $313k per year • Equity • Senior • In office (San Jose, United States) • Master's Degree • Full-Time • Senior
Python
AI/ML
Multimodal AI
PyTorch
FSDP
Diffusion Models
TensorRT
vLLM
Apply
Top 25% payVerified live · 1 day ago 6 days ago

Principal Scientist, ML – Overall Architect

$206k – $379k per year • Equity • Lead • In office (San Jose, United States) • PhD • Full-Time • Architect
PyTorch
FSDP
Computer Vision
Multimodal AI
Apply
Top 25% payVerified live · 1 day ago 22 days ago

Senior Machine Learning Engineer, AI Platform

$152k – $265k per year • Equity • Senior • In office (San Jose, United States) • Full-Time • Senior
C++
Go
Java
Python
Rust
C++
PyTorch C++
AI/ML
DeepSpeed
LLM
PyTorch
Ray
Ray Serve
TensorRT
TensorRT-LLM
Triton
vLLM
FSDP
DevOps
Kubernetes
Apply
Report

Wayve

wayve.ai
Wayve is a British autonomous driving technology company that develops end-to-end "Embodied AI" foundation models for self-driving vehicles and advanced driver assistance systems (ADAS). Founded in 2017 by University of Cambridge researchers, Wayve pioneered what it calls the "AV2.0" approach to autonomous vehicles. Unlike traditional self-driving stacks that rely on rigid hand-coded rules, high-definition (HD) 3D maps, and specialized sensors, Wayve uses a unified deep learning model.
wayve.ai • HQ: London, United Kingdom • Transportation & Logistics • Ride Hailing • Artificial Intelligence • Autonomous Driving • 201-500 employees
HQ: London, United Kingdom • Transportation & Logistics • Ride Hailing • Artificial Intelligence • Autonomous Driving • 201-500 employees
Verified live · 1 hour ago 6 days ago

Software Engineer

$130k – $293k per year (Estimated)In office (London, United Kingdom)
Python
AI/ML
Ignite
Multimodal AI
PyTorch
PyTorch Lightning
TensorFlow
Ray
Flyte
FSDP
Metaflow
DevOps
Azure
Argo Workflows
Datadog
Docker
Grafana
Kubernetes
OpenTelemetry
Prometheus
Terraform
Apply
Top 25% payVerified live · 1 hour ago 3 months ago

Research Scientist, Wayve Labs

$230k – $380k per year • Equity • Middle • 3+ years expIn office (Sunnyvale, United States) • Relocation • Visa sponsorship • Master's Degree • Middle
Python
AI/ML
Computer Vision
Ignite
Multimodal AI
PyTorch
Reinforcement Learning
RLHF
Synthetic Data
Mixture of Experts
DeepSpeed
JAX
FSDP
Robotics
Reinforcement Learning
Sim-to-Real
Apply
Verified live · 1 hour ago 3 months ago

Research Scientist, Wayve Labs

$143k – $303k per year (Estimated) • Middle • 3+ years expRemote/Hybrid (Vancouver, Canada) • Relocation • Visa sponsorship • Master's Degree • Middle
Python
AI/ML
Computer Vision
Ignite
Multimodal AI
PyTorch
Reinforcement Learning
RLHF
Synthetic Data
Mixture of Experts
DeepSpeed
JAX
FSDP
Robotics
Reinforcement Learning
Sim-to-Real
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 125 daysVerified live · 1 day ago 4 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

Harvard University

harvard.edu
Harvard University is a prestigious Ivy League private research institution based in Cambridge, Massachusetts, consistently ranked among the world's most influential universities. Founded in 1636, it is the oldest institution of higher learning in the United States and is renowned for its vast academic programs, rigorous research, and illustrious alumni. Today, the university continues to be a global leader in education and scholarship, maintaining a massive library system and an extensive network of specialized graduate and professional schools.
harvard.edu • HQ: Cambridge, United States • Education • Higher Education
HQ: Cambridge, United States • Education • Higher Education
Verified live · 1 hour ago 7 days ago

Senior Research Software Engineer

$143k – $258k per year (Estimated) • Senior • 7+ years expRemote/Hybrid (Cambridge, United States) • Visa sponsorship • Full-Time • Senior
C++
Python
C++
LLVM
PyTorch C++
TensorFlow C++
AI/ML
CUDA
CUDA Toolkit
DeepSpeed
JAX
ONNX
OpenCL
OpenMP
PyTorch
TensorFlow
TensorRT
Triton
cuDNN
FSDP
ROCm
DevOps
AWS
Git
HPC
Cybersecurity
HIPAA
Apply
Report

Synthesia

synthesia.io
Synthesia is an AI video generation platform that enables users to create professional, studio-quality videos from plain text scripts without needing traditional cameras or film crews. The platform utilizes hyper-realistic digital avatars and natural synthetic voice models to deliver localized video content across dozens of languages in minutes. Widely adopted by corporate enterprises, Synthesia streamlines corporate training, customer onboarding, and internal communications at scale.
synthesia.io • HQ: London, United Kingdom • Software • Artificial Intelligence • Generative Video • Video & Image Processing Software • 51-200 employees • Est. 2017
HQ: London, United Kingdom • Software • Artificial Intelligence • Generative Video • Video & Image Processing Software • 51-200 employees • Est. 2017
Verified live · 11 hours ago 8 days ago

Senior Applied Research Engineer - Video

$93k – $236k per year (Estimated) • Senior • Remote (Europe) • Full-Time • Senior
Python
AI/ML
DeepSpeed
Diffusion Models
PyTorch
FSDP
CUDA
CUDA Toolkit
DevOps
CI/CD
Datadog
AWS
Docker
SLURM
GitHub
Design
Webflow
Management
Miro
Apply
Report

The True Engineer

thetrueengineer.com
The True Engineer is an engineering leadership and software development publication platform hosted on Substack. Authored by Adlet Balzhanov, a senior software engineering lead with background in Big Tech and high-growth enterprise scale-ups, the publication delivers monthly long-form analytical essays, career frameworks, and strategic insights for software developers, engineering managers, and technology executives. Operating under a creator-driven digital subscription model, the newsletter focuses on technical leadership tactics, staff-level engineering practices, organizational dynamics within large tech companies, and navigating the operational impacts of AI on software development.
thetrueengineer.com • HQ: Stockholm, Sweden • Artificial Intelligence
HQ: Stockholm, Sweden • Artificial Intelligence
Verified live · 3 hours ago 8 days ago

ML Systems Performance Engineer (MFU)

$22k – $54k per year (Estimated) • Equity • In office (Almaty, Kazakhstan) • Relocation • Full-Time
C
C
MPI
AI/ML
CUDA
CUDA Toolkit
DeepSpeed
PyTorch
Triton
FSDP
Megatron-LM
Multimodal AI
Reinforcement Learning
InfiniBand
NCCL
NVLink
Mixture of Experts
Apply
Report

Innovaccer

innovaccer.com
Innovaccer is a healthcare technology company headquartered in San Francisco, California, and founded in 2014. The company provides a healthcare intelligence platform that unifies clinical, financial, and operational data to enable AI-powered workflows, population health management, and revenue cycle optimization. It serves health systems, payers, and life sciences organizations across the United States and internationally, utilizing autonomous AI agents to improve patient outcomes and administrative efficiency.
innovaccer.com • HQ: San Francisco, United States • Business Process Automation (BPA) • Software • Analytics Platforms • Medical AI • Artificial Intelligence • Health Care • Data & Analytics • Digital Health • 501-1000 employees • Est. 2014
HQ: San Francisco, United States • Business Process Automation (BPA) • Software • Analytics Platforms • Medical AI • Artificial Intelligence • Health Care • Data & Analytics • Digital Health • 501-1000 employees • Est. 2014
Verified live · 1 day ago 8 days ago

Senior Artificial Intelligence Researcher

$163k – $317k per year (Estimated) • Senior • In office (San Francisco, United States) • Bachelor's Degree • Full-Time • Senior
Python
AI/ML
AI Agents
DeepSpeed
Fine-tuning
LLM
PyTorch
RAG
SGLang
vLLM
FSDP
Hugging Face
Apply
Verified live · 1 day ago 8 days ago

Principal AI Researcher

$164k – $352k per year (Estimated) • Lead • In office (San Francisco, United States) • Bachelor's Degree • Full-Time • Principal
Python
AI/ML
AI Agents
DeepSpeed
Fine-tuning
LLM
PyTorch
RAG
SGLang
vLLM
FSDP
Hugging Face
Post-training
Apply
Verified live · 1 day ago 8 days ago

Artificial Intelligence Researcher

$158k – $345k per year (Estimated)In office (San Francisco, United States) • Bachelor's Degree • Full-Time
Python
AI/ML
AI Agents
DeepSpeed
Fine-tuning
LLM
PyTorch
RAG
SGLang
vLLM
FSDP
Hugging Face
Apply
Report

NVIDIA

nvidia.com
NVIDIA is an American technology company founded in 1993 that invented the graphics processing unit and has become the dominant supplier of accelerated computing platforms for artificial intelligence. Its portfolio spans data centre GPUs and systems built on the Hopper and Blackwell architectures, GeForce consumer graphics, automotive and robotics platforms, high-speed networking acquired with Mellanox, and the CUDA software stack that binds the ecosystem together. Headquartered in Santa Clara, California, the company sells to cloud providers, enterprises, research institutions and gamers worldwide and is one of the most valuable listed businesses on the Nasdaq.
nvidia.com • HQ: Santa Clara, United States • AI Infrastructure • Semiconductors • AI Agents • Big Data • LLM & Generative AI • Machine Learning • Hardware • Data & Analytics • Information Technology • Artificial Intelligence • 5000+ employees • Est. 1993
HQ: Santa Clara, United States • AI Infrastructure • Semiconductors • AI Agents • Big Data • LLM & Generative AI • Machine Learning • Hardware • Data & Analytics • Information Technology • Artificial Intelligence • 5000+ employees • Est. 1993
$184k – $288k per year • Senior • 5+ years expIn office (Santa Clara, United States) • Master's Degree • Full-Time • Senior
C++
Python
C++
PyTorch C++
AI/ML
LLM
PyTorch
Ray
Reinforcement Learning
RLHF
AI Agents
DPO
FSDP
GRPO
Post-training
PPO
DeepSpeed
DeepSpeed-Chat
Multimodal AI
Quantization
SGLang
TensorRT
TensorRT-LLM
vLLM
VLM
InfiniBand
Megatron-LM
Mixture of Experts
NCCL
NVIDIA NeMo
NVLink
Pre-training
DevOps
Kubernetes
Apply
$272k – $431k per year • Lead • 15+ years expIn office (Santa Clara, Austin, United States) • Bachelor's Degree • Full-Time • Principal
Python
AI/ML
CUDA
CUDA Toolkit
DeepSpeed
LLM
SGLang
TensorRT
TensorRT-LLM
vLLM
FSDP
Megatron-LM
NCCL
NVLink
DevOps
HPC
Apply
$124k – $196k per year • Remote (United States) • Visa sponsorship • Master's Degree
Python
AI/ML
JAX
PyTorch
FSDP
InfiniBand
NVIDIA NeMo
DevOps
AWS
Azure
Docker
GCP
Kubernetes
SLURM
HPC
Apply
Report

Roblox Corporation

roblox.com
Roblox Corporation (NYSE: RBLX) is an immersive global online gaming and user-generated content (UGC) platform headquartered in San Mateo, California. Founded in 2004 by David Baszucki and Erik Cassel, the company operates an end-to-end 3D digital ecosystem consisting of three core components: Roblox Client (the application through which users explore virtual experiences), Roblox Studio (the proprietary toolset that allows creators and developers to build 3D worlds), and Roblox Cloud (the infrastructure that powers real-time services and multiplayer hosting).
roblox.com • HQ: San Mateo, United States • Artificial Intelligence • LLM & Generative AI • Gaming • Metaverse • Game Development • 1001-5000 employees • Est. 2004
HQ: San Mateo, United States • Artificial Intelligence • LLM & Generative AI • Gaming • Metaverse • Game Development • 1001-5000 employees • Est. 2004
Open 310 daysVerified live · 9 min ago 10 months ago

Principal Machine Learning Infrastructure Engineer, Ads & Discovery

$153k – $328k per year (Estimated) • Equity • Lead • 5+ years expRemote/Hybrid (San Mateo, United States) • Visa sponsorship • Bachelor's Degree • Full-Time • Principal
CUDA Toolkit
Quantization
SGLang
vLLM
FSDP
AI Agents
Apply
Report

LiveKit

livekit.io
LiveKit is an open-source real-time audio, video, and AI voice infrastructure platform headquartered in San Francisco. The company provides cloud and developer infrastructure designed to build ultra-low-latency video conferencing, live streaming, and interactive multimodal AI voice agents into web and mobile applications. Powered by flexible SDKs and scalable server orchestration, LiveKit is widely used by developers and enterprise teams to deploy real-time voice and video applications at global scale.
livekit.io • HQ: San Francisco, San Jose, United States • Artificial Intelligence • AI Agents • Speech & Audio AI • Est. 2021
HQ: San Francisco, San Jose, United States • Artificial Intelligence • AI Agents • Speech & Audio AI • Est. 2021
Top 25% payVerified live · 2 hours ago 13 days ago

Research Engineer (Reinforcement Learning)

$135k – $300k per year • Equity • Remote (EMEA, North America) • Full-Time
Python
AI/ML
Synthetic Data
PEFT
LiveKit
OpenAI
Post-training
Fine-tuning
Llama
LoRA
Qwen
Reinforcement Learning
SGLang
vLLM
FSDP
GRPO
LLM Evaluation
Marketing
Salesforce
Apply
Report

Databricks

databricks.com
Databricks is an American software company founded in 2013 by the original creators of Apache Spark, Delta Lake and MLflow at the University of California, Berkeley. Its data intelligence platform unifies data engineering, warehousing, governance, analytics and machine learning on a lakehouse architecture that stores open table formats in a customer's own cloud object storage. Headquartered in San Francisco and available on AWS, Azure and Google Cloud, the platform is used by tens of thousands of organisations to build pipelines, dashboards and production AI applications.
databricks.com • HQ: San Francisco, United States • MLOps • Machine Learning • Big Data • Data Engineering • Artificial Intelligence • Data & Analytics • Data Warehousing • 5000+ employees • Est. 2013
HQ: San Francisco, United States • MLOps • Machine Learning • Big Data • Data Engineering • Artificial Intelligence • Data & Analytics • Data Warehousing • 5000+ employees • Est. 2013
Verified live · 1 hour ago 2 months ago

Staff Software Engineer, AI Runtime

$220k – $412k per year (Estimated) • Staff+ • 10+ years expIn office (Mountain View, United States) • Bachelor's Degree • Staff
Databricks
AI/ML
DeepSpeed
Fine-tuning
PyTorch
FSDP
InfiniBand
Megatron-LM
NVLink
Pre-training
Apply
Verified live · 1 hour ago 2 months ago

Senior Software Engineer, AI Runtime

$183k – $355k per year (Estimated) • Senior • 5+ years expIn office (Mountain View, United States) • Bachelor's Degree • Senior
Databricks
AI/ML
DeepSpeed
Fine-tuning
PyTorch
FSDP
InfiniBand
Megatron-LM
NVLink
Pre-training
Apply
Verified live · 1 hour ago 1 month ago

Sr. Engineering Manager, AI Runtime

$210k – $403k per year (Estimated) • Lead • 8+ years expIn office (Mountain View, United States) • Bachelor's Degree • Senior
Databricks
AI/ML
DeepSpeed
Fine-tuning
LLM
PyTorch
FSDP
Megatron-LM
NCCL
Apply
Report

Nimble Robotics

nimble.ai
Nimble (Nimble Robotics) is an AI-powered robotics and autonomous third-party logistics (3PL) company headquartered in San Francisco, California. Founded in 2017 by Simon Kalouche, the company develops fully autonomous fulfillment networks powered by intelligent picking, packing, and sorting robots ("Superhumanoids") alongside cloud supply chain management software.
nimble.ai • HQ: San Francisco, United States • Transportation & Logistics • Commerce • Logistics • 51-200 employees • Est. 2017
HQ: San Francisco, United States • Transportation & Logistics • Commerce • Logistics • 51-200 employees • Est. 2017
Top 25% payVerified live · 9 hours ago 13 days ago

Senior/Staff Software Engineer, Infrastructure (ML)

$210k – $300k per year • Staff+ • 4+ years expIn office (San Francisco, United States) • Bachelor's Degree • Staff
C++
Python
Rust
C++
PyTorch C++
AI/ML
CUDA Toolkit
JAX
PyTorch
DeepSpeed
FSDP
Megatron-LM
NCCL
DevOps
Kubernetes
Apply
Report

Lightning AI

lightning.ai
Lightning AI is an artificial intelligence technology company headquartered in New York City, New York, and established in 2019. The organization provides a unified platform and the open-source PyTorch Lightning framework to facilitate the building, training, and deployment of machine learning models. It operates globally by offering scalable cloud infrastructure and development tools to individual researchers and enterprise teams across various industries.
lightning.ai • HQ: New York, United States • Cloud Management • Software • Cloud Computing • Information Technology • MLOps • Machine Learning • Artificial Intelligence • Est. 2019
HQ: New York, United States • Cloud Management • Software • Cloud Computing • Information Technology • MLOps • Machine Learning • Artificial Intelligence • Est. 2019
Open 789 daysVerified live · 2 hours ago 2 years ago

Research Engineer, AI/ML Systems

$105k – $221k per year (Estimated) • Equity • Remote/Hybrid (London, United Kingdom) • Bachelor's Degree
CUDA Toolkit
PyTorch
PyTorch Lightning
SGLang
vLLM
CUDA
DeepSpeed
Triton
FSDP
Hugging Face
Post-training
Apply
Verified live · 2 hours ago 19 days ago

Senior Research Engineer, LLM Training & Post-Training

$152k – $295k per year (Estimated) • Equity • Senior • Remote/Hybrid (New York, United States) • Master's Degree • Senior
Python
AI/ML
CUDA Toolkit
DeepSpeed
Fine-tuning
Lightning Fabric
LLM
PEFT
PyTorch
PyTorch Lightning
Reinforcement Learning
RLHF
SGLang
TensorRT
Transformers
TRL
vLLM
DPO
FSDP
GRPO
Hugging Face
Megatron-LM
Post-training
PPO
SFT
DevOps
Platform Engineering
Apply
Report

Mirage

mirage.app
Mirage is an artificial intelligence technology enterprise that develops generative media models and automated video creation tools. Headquartered in New York City, New York, the company provides an integrated platform featuring AI-generated avatars, automated captioning, dynamic audio synthesis, and layered video editing. By powering multimodal content generation through both user-facing applications and scalable APIs, its technology enables creators, marketing teams, and developers to streamline high-quality video production.
mirage.app • HQ: New York, United States • Software • Video & Image Processing Software • Artificial Intelligence • Generative Video
HQ: New York, United States • Software • Video & Image Processing Software • Artificial Intelligence • Generative Video
Top 25% payVerified live · 1 hour ago 15 days ago

Research Engineer, Generative Video

$175k – $275k per year • Junior • 2+ years expIn office (New York, United States) • Bachelor's Degree • Full-Time • Junior
CUDA
CUDA Toolkit
Knowledge Distillation
Multimodal AI
PyTorch
Quantization
Triton
FSDP
Apply
Report

UiPath

uipath.com
UiPath is a global enterprise software company that leads the market in Robotic Process Automation (RPA) and AI-driven automation tools. Its platform enables organizations to build, deploy, and manage software robots that automate repetitive digital tasks and complex business processes across legacy and modern applications. Headquartered in New York City, the company helps enterprises drive operational efficiency, improve accuracy, and accelerate digital transformation worldwide.
uipath.com • HQ: New York, United States • Artificial Intelligence • AI Agents • Software • Business Process Automation (BPA) • 1001-5000 employees • Est. 2005
HQ: New York, United States • Artificial Intelligence • AI Agents • Software • Business Process Automation (BPA) • 1001-5000 employees • Est. 2005
Top 25% payVerified live · 1 hour ago 17 days ago

Principal Applied Scientist- AI

$218k – $265k per year • Lead • In office (Bellevue, United States) • PhD • Full-Time • Principal
Python
AI/ML
AI Agents
Function Calling
LLM
Multimodal AI
PyTorch
RAG
Reinforcement Learning
RLHF
Synthetic Data
LangChain
DPO
GRPO
Post-training
PPO
CUDA Toolkit
DeepSpeed
Fine-tuning
JAX
LangGraph
Ray
Semantic Kernel
vLLM
FSDP
Hugging Face
Model Context Protocol
DevOps
Azure
Kubernetes
Robotics
Reinforcement Learning
Management
UiPath
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.