87 jobs for FSDP

Generative AI Engineer, Senior Staff/Manager (LLM/VLM) USD 140K-211K

DeepSpeed | Distributed Training | FSDP | Knowledge Distillation | Language Models

Senior-level Full Time

Ho Chi Minh, Hồ Chí Minh, …

2d ago

Foundation Model Engineer USD 200K-230K

DPO | Data Generation | Dataset Distillation | Distributed Training | Evaluation methodology

Senior-level Full Time

United States - Remote R

2d ago

LLM Engineer USD 100K-150K

Adapter methods | Attention Optimization | Automated benchmarking | DPO | DeepSpeed ZeRO

Senior-level Full Time

United States - Remote R

3d ago

AI Developer Technology Intern, Robotics - 2027 CNY 28K-50K

C++ | CUDA | DeepSpeed | Distributed Training | FSDP

Entry-level Internship

China, Shanghai

4d ago

Senior Applied Scientist / Engineer, Training & Inference USD 216K-313K

Cost Optimization | Distributed Training | Execution orchestration | FSDP | GPU efficiency

Senior-level Full Time

San Jose, United States R

4d ago

Machine Learning Engineer 3 INR 1500K-3000K

Attention | Automated retraining | CI/CD | Contrastive Learning | DDP

Mid-level Full Time

Noida, India R

4d ago

Machine Learning Engineer INR 1860K-3000K

Apache Spark | Attention Mechanisms | Automated retraining | CI/CD | Contrastive Learning

Mid-level Full Time

Bangalore, India R

4d ago

Machine Learning Engineer 4 INR 3242K-5000K

Apache Spark | Attention | Automated retraining | CI/CD | Checkpointing

Senior-level Full Time

Noida, India R

4d ago

Distributed Training & Inference Optimization Engineer (LLM) - GPU Optimization Department (GPUOD) JPY 8000K-10800K

CUDA | Continuous batching | DeepSpeed | Dynamic batching | FSDP

Mid-level Full Time

Rakuten Crimson House, Japan

4d ago

Machine Learning/AI Scientist PhD Intern, Winter 2027 USD 80K-220K

C++ | Causal Discovery | Causal Inference | Computer Vision | DDP

401k retirement plan | Health plans | Mental health support | Onsite work | Paid internship

Entry-level Full Time Internship

Los Gatos, United States

4d ago

Large Language Model Specialist USD 100K-150K

Attention Optimization | Benchmarking | DPO | DeepSpeed ZeRO | Evaluation methodology

Mid-level Full Time

Sammamish, WA R

4d ago

Generative AI Engineer USD 90K-115K

DPO | Data Generation | Dataset Distillation | DeepSpeed ZeRO | Distributed Training

Senior-level Full Time

United States - Remote R

4d ago

AI Infrastructure Engineer - Recommendation & LLM USD 160K-220K

Algorithms | C++ | CUDA | CUDA graph | Computer Architecture

Senior-level Full Time

San Jose, California, United States

5d ago

LLM Engineer USD 72K-100K

DPO | Distributed Training | Evaluation | FSDP | GPU clusters

Career growth | Remote work

Senior-level Full Time

United States - Remote R

5d ago

具身大模型训练框架工程师/专家 CNY 240K-480K

CUDA | Checkpointing | Collective communication | Computer Vision | Data parallelism

Senior-level Full Time

上海

6d ago

Principal AI Researcher USD 170K-230K

AI Agents | Ablation Studies | DeepSpeed | Distributed Training | Efficient Fine Tuning

Basic life insurance | Dental benefits | Generous PTO | Legal aid | Long-term disability

Mid-level Full Time

San Francisco, California, United States

6d ago

Artificial Intelligence Researcher USD 170K-241K

AI Agents | Ablation Studies | DeepSpeed | Experiment design | FSDP

Dental insurance | Legal aid | Life insurance | Long-term disability | Medical insurance

Mid-level Full Time

San Francisco, California, United States

6d ago

NLP AI Engineer USD 130K-180K

Data Generation | DeepSpeed ZeRO | Direct Preference Optimization | Distributed Training | Docker

Senior-level Full Time

United States - Remote R

6d ago

Senior Machine Learning Engineer - ML Training Infrastructure USD 170K-240K

AWS | Azure | Debugging | Distributed Computing | FSDP

Health and wellbeing programs | Medical/Dental/Vision | Paid Vacation & Holidays | Relocation benefits | Tuition assistance

Senior-level Full Time

GM Automation - Sunnyvale - GM … R

6d ago

Principal Software Engineer – PyTorch Training Frameworks USD 100K-150K

C++ | CUDA | DDP | Distributed Systems | Distributed Training

Senior-level Full Time

US,CA,San Jose

6d ago

Software Engineer, SystemML - AI Networking USD 124K-220K

C# | C++ | CUDA | CUDA programming | Data-parallel

Entry-level Full Time

Menlo Park, CA

8d ago

Co-op Researcher – LLM Post-Training & Agentic Coding CAD 58K-104K

DeepSpeed | Distributed Systems | Docker | FSDP | Fine Tuning

Entry-level Full Time

Edmonton, Alberta, Canada

9d ago

Senior LLM Engineer USD 195K-255K

AI Foundry | AI Search | Azure AI | Azure AI Foundry | Azure AI Search

Senior-level Full Time

Austin, Texas | Fremont, California | …

9d ago

Senior Software Engineer, RL Post-Training Frameworks USD 184K-287K

Actor Based Programming | C# | C++ | Deep learning | Distributed Systems

Senior-level Full Time

US, CA, Santa Clara R

9d ago

Software Engineer III - Machine Learning Platform USD 170K-186K

AWS EKS | Alerting | Amazon Web Services | CI/CD | Checkpointing

Backup childcare | Financial coaching | Health care | Mental health support | On-site health and wellness centers

Senior-level Full Time

Palo Alto, CA, United States

9d ago

Senior LLM Engineer USD 195K-255K

AI Foundry | AI Search | Azure AI | Azure AI Foundry | Azure AI Search

Senior-level Full Time

Austin, Texas | Fremont, California | …

10d ago

Research Scientist - Robot Learning (VLA / WAM) GBP 200K-300K

Action Representation | Calibration | Computer Vision | Control | Diffusion Models

Executive-level Full Time

London

10d ago

Tech Lead, Machine Learning Engineer - Global E-Commerce (Conversational AI) SGD 150K-199K

AI Feedback | Agent systems | Context engineering | Deep learning | DeepSpeed

Senior-level Full Time

Singapore, Singapore

11d ago

Student Researcher (AI Foundation Models Infrastructure - Seed Infra) - 2026 Start (PhD) USD 218K-387K

C++ | CUDA | Distributed Computing | Distributed Systems | Distributed Training

Health insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick time

Entry-level Full Time

San Jose, California, United States

11d ago

Student Researcher (Seed - LLM - Model) - 2026 Start (PhD) USD 170K-170K

Algorithms | C++ | Data Structures | Deep learning | DeepSpeed

Health insurance | Housing allowance | Life insurance | Paid Holidays | Paid sick time

Entry-level Full Time

San Jose, California, United States

11d ago

Research Scientist - LLM Training System as a Service - Global Frontier Tech Recruitment Program - 2027 Start (PhD) USD 218K-480K

Access Optimization | Algorithms | CUDA | Computer Architecture | Data Structures

401k match | Life insurance | Long-term disability | Medical/Dental/Vision insurance | Paid Holidays

Entry-level Full Time

San Jose, California, United States

11d ago

Applied Scientist - LLM Training System as a Service - Global Frontier Tech Recruitment Program - 2027 Start (PhD) USD 218K-480K

Algorithms | CUDA | Computer Architecture | Data Structures | Deep learning

401k match | Dental insurance | Life insurance | Long-term disability | Medical insurance

Entry-level Full Time

San Jose, California, United States

11d ago

Research Engineer - LLM Training Infrastructure - Seed Infra USD 241K-456K

C++ | CUDA | Checkpointing | Data parallelism | Distributed Training

401k match | Dental insurance | Health insurance | Life insurance | Long-term disability

Senior-level Full Time

Seattle, Washington, United States

11d ago

Principal Machine Learning Engineer

Apache Arrow | Apache Spark | DPO | Deep learning | DeepSpeed

Senior-level Full Time

Seoul, Korea R

15d ago

Principal Machine Learning Engineer SGD 160K-225K

Apache Arrow | Apache Spark | DPO | Deep learning | DeepSpeed

Senior-level Full Time

Singapore R

15d ago

Principal Applied Scientist- AI USD 218K-265K

Azure AI | CUDA | DPO | Data Generation | DeepSpeed

Senior-level Full Time

Bellevue R

16d ago

Helix AI Engineer, Training Performance USD 200K-400K

Activation checkpointing | C++ | CUDA | FSDP | Infiniband

Mid-level Full Time

San Jose, CA

16d ago

Tech Lead, Robotic AI Model USD 150K-180K

Action Tokenization | Behavior Cloning | Data Augmentation | Data Quality | Data Quality Filtering

Senior-level Full Time

El Segundo, California, United States

17d ago

Senior Machine Learning Engineer – Predictive World Model USD 174K-295K

Computer Vision | Data Processing | Deep learning | DeepSpeed | Diffusion Models

Computational resources | Dinners | Fun activities | In house infrastructure | Lunches

Senior-level Full Time

Santa Clara, CA

17d ago

Senior Research Engineer, LLM Training & Post-Training USD 165K-310K

CUDA | DPO | DeepSpeed | Distributed Training | Experiment design

401k matching | Company holidays | Comprehensive health coverage | Dental coverage | Flexible time off

Senior-level Full Time

New York, New York, United States; … R

17d ago

机器学习平台研发工程师/专家 CNY 240K-360K

DDP | Distributed Training | Docker | FSDP | Go

Senior-level Full Time

北京、上海

19d ago

Machine Learning Engineer INR 3470K-5000K

Apache Spark | Attention Mechanisms | Automated retraining | CI/CD | Checkpointing

Senior-level Full Time

Noida, India R

19d ago

【27届校招】大模型训练推理框架工程师 CNY 500K-500K

Airflow | C++ | CI/CD | CUDA | DeepSpeed

Mid-level Full Time

深圳、北京

19d ago

Lead Software Engineer - Machine Learning Platform USD 170K-195K

AWS | Airflow | Alerting | Argo Workflows | Artifact management

Senior-level Full Time

Palo Alto, CA, United States

20d ago

Principal Machine Learning Engineer CNY 144K-240K

Apache Arrow | Apache Spark | DPO | Deep learning | DeepSpeed

Senior-level Full Time

China R

20d ago

Artificial Intelligence Engineer USD 143K-190K

AI Agents | Ablation Studies | DeepSpeed | Efficient Fine Tuning | Experiment design

Basic life insurance | Dental insurance | Disability insurance | Legal aid | Medical insurance

Mid-level Full Time

San Francisco, California, United States

20d ago

Staff ML Engineer, Perception Research USD 251K-310K

3D Perception | Autoregressive models | Data Pipelines | Data parallelism | Diffusion Models

401k match | Baby bonding leave | Dental insurance | Disability insurance | Health insurance

Senior-level Full Time

Mountain View, CA, USA ; New … R

22d ago

Recommendation Architecture AI/ML Infrastructure Engineer Intern (Data-Arch-TikTok Live) - 2027 Summer USD 119K-177K

Algorithms | C++ | CUDA | Computer Systems | Continuous batching

Development workshops | Hands-on industry experience | Learning programs | Social events

Entry-level Internship

San Jose, California, United States

23d ago

Machine Learning Engineer – Digital Intelligence USD 170K-201K

Ablation Studies | Attention Mechanisms | Deep learning | DeepSpeed | Distributed Training

Backup childcare | Financial coaching | Health care coverage | Mental health support | Onsite wellness centers

Senior-level Full Time

Palo Alto, CA, United States

23d ago

Senior Machine Learning Engineer – Digital Intelligence USD 166K-201K

Ablation Studies | Attention Mechanisms | Benchmarking | DeepSpeed | Distributed Training

Backup childcare | Financial coaching | Health care coverage | Mental health support | On-site health and wellness centers

Senior-level Full Time

Palo Alto, CA, United States

24d ago

Find jobs in AI/ML, Data Science and Big Data