Distributed training Jobs

Python Machine learning Deep learning Distributed training

If you’ve never thought about a career in finance, you’re in good company. Many of us were in the same position before working here. If you have a curious…

Machine Learning Researcher

Jane Street

Hong Kong, Hong Kong8d ago

Python Machine learning Deep learning Distributed training

If you’ve never thought about a career in finance, you’re in good company. Many of us were in the same position before working here. If you have a curious…

Machine Learning Researcher

Jane Street

New York, New York, United States8d ago

Python Machine learning Deep learning Distributed training

If you’ve never thought about a career in finance, you’re in good company. Many of us were in the same position before working here. If you have a curious…

2026 Summer Intern, PhD, ML Training Infra and Efficiency

Waymo

Mountain View, CA, USA8d ago

Intern

Python C++Machine learning Deep learning TensorFlow PyTorch Distributed training

Python or C++ experience Scale model training to multiple accelerators Reinforcement learning

Senior Machine Learning Infrastructure Engineer, Simulation

Waymo

Mountain View, CA8d ago

Senior

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,…

Manager I, Engineering - AI Platform - Training & Serving

Datadog

Datadog Mentoring Code review Distributed training

Previous experience (1+ years) leading software engineering teams, as a tech lead or people manager Strong technician with a mix of backend, data engineer and…

Quantitative Researcher

Jane Street

Python Linear Machine learning Deep learning Distributed training

If you’ve never thought about a career in finance, you’re in good company. Many of us were in the same position before working here. If you have a curious…

Research Engineer/Research Scientist, Audio

Anthropic

San Francisco, CA8d ago

Kubernetes Machine learning PyTorch LLM Large language models Fine-tuning Inference optimization Distributed training AI safety

Large language model pretraining and finetuning Training diffusion models for image and audio generation Reinforcement learning for large language models and…

Machine Learning Engineer

IMC

Chicago, United States8d ago

Python C++Machine learning TensorFlow PyTorch A/B testing TensorRT Distributed training CUDA GPU acceleration

5+ years of experience in machine learning with a focus on training or inference systems Hands-on experience with real-time, low-latency ML pipelines in…

Senior Reinforcement Learning Engineer

Apptronik

Python C++PyTorch Computer vision Mentoring Code review Distributed training

Deep, hands-on expertise (5+ years) with common RL frameworks (e.g., PyTorch, JAX) and high-fidelity physics simulators (e.g., MuJoCo, IsaacGym) Mastery of…

Staff Research Engineer, Pre-training Science

Remote - United States8d ago

Amsterdam, Netherlands; Chicago, United States; Hong Kong, Hong Kong; London, United Kingdom; New York, United States; Sydney, Australia8d ago

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.…

Principal Machine Learning Engineer

IMC

Python C++Machine learning Deep learning TensorFlow PyTorch Mentoring Transformers TensorRT Distributed training+1

8+ years of experience building ML platforms or infrastructure at a leading tech company, research lab, or quantitative firm A track record of designing and…

Special Projects - OCTO

Moveworks

Mountain View, CA8d 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,…

Machine Learning Researcher

Jane Street

Python Machine learning Deep learning Distributed training

If you’ve never thought about a career in finance, you’re in good company. Many of us were in the same position before working here. If you have a curious…

Research Engineer - Pretraining

Anthropic

Python Kubernetes Machine learning Deep learning PyTorch Data pipelines Problem solving Distributed training AI safety

Advanced degree (MS or PhD) in Computer Science, Machine Learning, or a related field Strong software engineering skills with a proven track record of…

Staff Machine Learning Engineer

Waymo

Python Machine learning Deep learning TensorFlow LLM Large language models RLHF Distributed training

10+ years of relevant experience in ML/RL research and application. Experience in the autonomous vehicles domain, robotics, or complex simulation…

AI Research Engineer – Datadog AI Research (DAIR)

Datadog

New York, New York, USA8d ago

Sunnyvale, CA / Bellevue, WA8d ago

Python Rust C++Datadog Machine learning PyTorch LLM Security Mentoring Documentation+5

You have strong software engineering skills with experience in domains such as observability, SRE, or security You have depth in distributed computing and ML…

Staff Engineer - Perf and Benchmarking

CoreWeave

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…

Machine Learning Engineer

Waymo