DeepSpeed Jobs

Machine learning TensorFlow PyTorch Mentoring Ownership Transformers Distributed training DeepSpeed

10+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure such as developing, designing, scaling,…

Staff Research Engineer, Pre-training Science

Remote - United States8d ago

Staff

Python AWS CI/CD Machine learning Deep learning PyTorch LLM Mentoring Ownership Large language models+6

7+ years of experience in Machine Learning engineering or research, with a specific focus on LLM Pre-training, Domain Adaptation, or Transfer Learning.…

Special Projects - OCTO

Moveworks

Mid-level

Sunnyvale, CA / Bellevue, WA8d ago

Python JavaScript TypeScript AWS GCP Azure Kubernetes Docker Machine learning Deep learning+22

Experience in enterprise software environments and understanding of B2B customer needs Deep specialization in one or more AI domains: NLP, computer vision,…

Staff Engineer - Perf and Benchmarking

CoreWeave

Staff

Mountain View, CA, USA: San Francisco, CA, USA8d ago

Spring Kubernetes CI/CD Prometheus Grafana Machine learning PyTorch LLM Mentoring Ownership+8

10+ years building distributed systems or HPC/cloud services, with deep expertise on large-scale ML training or similar high-performance workloads. Proven…

Director Engineering, AI Training Platform

CoreWeave

Spring Kubernetes CI/CD Prometheus Grafana Machine learning PyTorch Data pipelines Mentoring OpenTelemetry+3

10+ years building large-scale distributed systems or HPC/cloud services, with 5+ years leading engineering teams. Proven success delivering mission-critical…

AI Research Scientist – Datadog AI Research (DAIR)

Datadog

Datadog Machine learning Deep learning TensorFlow PyTorch LLM NLP Security Mentoring Large language models+4

You hold a PhD in Computer Science, Machine Learning, or a related field, with deep expertise in areas like generative modeling, AI agents, reinforcement…

Senior Software Engineer, Simulation ML Infrastructure

Waymo

Machine learning TensorFlow PyTorch Ownership Transformers DeepSpeed

5+ years of professional software engineering experience, with at least 3 years in machine learning infrastructure such as developing, designing, scaling,…

AI Research Scientist – Datadog AI Research (DAIR)

Datadog

New York, New York, USA8d ago

Mid-level

Datadog Machine learning Deep learning TensorFlow PyTorch LLM NLP Security Mentoring Large language models+4

You hold a PhD in Computer Science, Machine Learning, or a related field, with deep expertise in areas like generative modeling, AI agents, reinforcement…

Engineering Manager - Machine Learning Infrastructure

Moveworks

Manager

Python Go C++Kubernetes Machine learning Deep learning PyTorch LLM Mentoring Ownership+6

As the Engineering Manager for the Machine Learning Infrastructure team, you will spearhead the development of the cutting-edge platform that powers…

Senior Machine Learning Engineer - I (MLOps/LLMOps)

Sumo Logic

United States (HQ)8d ago

Bengaluru, Karnataka, India8d ago

Python Java Rust REST gRPC AWS GCP Azure Kubernetes Docker+32

As a Senior Machine Learning Engineer - MLOps/LLMOps, you will design, build, and scale production-grade infrastructure and platforms that enable the full…

Lead Data Scientist

CommerceIQ

Lead

Remote - United States8d ago

AWS GCP Azure CI/CD Machine learning Deep learning TensorFlow PyTorch Spark NLP+10

CommerceIQ’s AI-powered digital commerce platform is revolutionizing the way brands sell online. Our unified ecommerce management solutions empower brands to…

Senior Research Engineer, Post-training & Evaluation

Python CI/CD Machine learning PyTorch LLM Ownership MLflow Large language models Transformers Instruction tuning+4

4+ years of professional experience in machine learning engineering, with a focus on LLM fine-tuning or evaluation. Fluency in Python and PyTorch, with…

Senior Cluster Site Reliability Engineer

Voleon

Python Ruby AWS GCP Kubernetes Docker Terraform Ansible S3 Prometheus+12

5+ years of experience in SRE or DevOps roles, preferably working as a senior engineer or tech lead Knowledge of HPC/batch compute frameworks (Slurm, Kueue,…

Staff Software Engineer AI

FourKites

Chennai or Remote, India8d ago

Staff

Bengaluru, Karnataka, India8d ago

Python FastAPI PostgreSQL Redis AWS Kubernetes Machine learning PyTorch LLM A/B testing+25

Technical Skills 8+ years software engineering, 3+ years in production AI/ML systems Expertise in Python, PyTorch/JAX, and AI frameworks (LangChain,…

Data Science- Manager

CommerceIQ

Manager

Machine learning TensorFlow PyTorch Mentoring Ownership Transformers Distributed training DeepSpeed

AWS GCP Azure CI/CD Machine learning Deep learning TensorFlow PyTorch Spark NLP+10

CommerceIQ’s AI-powered digital commerce platform is revolutionizing the way brands sell online. Our unified ecommerce management solutions empower brands to…

Staff Software Engineer, Simulation ML Infrastructure

Waymo

MS in Computer Science, Robotics, similar technical field of study, or equivalent practical experience 10+ years of professional software engineering…

Senior Machine Learning Engineer II - LLM

Moveworks

Bengaluru, Karnataka, India8d ago

Python Go C++Kubernetes Machine learning Deep learning PyTorch Data pipelines LLM NLP+8

We are looking for a Machine Learning Engineer to help build cutting edge ML infrastructure for building and serving LLM’s at Moveworks. This role will be…

Data Scientist II

CommerceIQ

Mid-level