"distributed training" Jobs - Joblaze

Showing 20 of 191 results

Join Databricks as a Staff Software Engineer to drive the architecture of a managed GPU training platform for large-scale AI models.

Databricks Mountain View, California; San Francisco, California $190k–$265k/yr Published 6 days ago

NVLink Megatron DeepSpeed PyTorch GPU

Apply →

Senior Software Engineer, AI Runtime

Join Databricks as a Senior Software Engineer to build and scale a managed GPU training platform for AI models.

Databricks Mountain View, California; San Francisco, California $160k–$225k/yr Published 6 days ago

NVLink Megatron DeepSpeed PyTorch GPU

Flexible on stack 70% coding

Apply →

Senior Engineering Manager, AI Runtime

Databricks Mountain View, California; San Francisco, California $228.6k–$314.3k/yr Published 2 months ago

DeepSpeed AI/ML Composer Megatron-LM PyTorch

Apply →

Member of Technical Staff (AI Infrastructure Engineer)

Perplexity San Francisco Published 2 months ago

C++ AWS Slurm PyTorch Python

Apply →

Research Engineer, Pretraining Scaling Visa

Anthropic San Francisco, CA $350k–$850k/yr Published 8 months ago

ML frameworks JAX model operations AI/ML large-scale distributed systems

Apply →

Staff Engineer, Distributed Storage and HPC & AI Infrastructure

Design and deliver multi-petabyte storage systems for AI workloads at Together AI, optimizing performance and cost.

Together AI San Francisco $250k–$300k/yr Published 1 week ago

Terraform iSCSI Grafana RAID Prometheus

Flexible on stack

Apply →

Senior Software Engineer - Distributed Data Systems

Databricks San Francisco, California $166k–$225k/yr Published 6 years ago

C++ Java Hadoop Apache Spark Scala

Apply →

Staff Software Engineer - Distributed Data Systems

Databricks San Francisco, California $192k–$260k/yr Published 3 years ago

C++ Delta Lake Java Hadoop Apache Spark

Apply →

Staff Backend Software Engineer- (AI Platform)

Databricks San Francisco, California $192k–$260k/yr Published 2 months ago

SGLang vLLM AI/ML APIs GPU

Apply →

Research Engineer, Production Model Post-Training Remote Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $350k–$500k/yr Published 1 year ago

LLMs deep learning frameworks Constitutional AI Python distributed computing

Apply →

Staff Machine Learning Platform Engineer Remote

Faire San Francisco, CA $224k–$308k/yr Published 3 months ago

Terraform Datadog Sagemaker AWS Cockroach DB

Apply →

Staff Software Engineer, Foundational Model Serving

Databricks San Francisco, California $192k–$260k/yr Published 7 months ago

SGLang vLLM AI/ML GPU

Apply →

Senior GenAI Research Engineer - Optimization and Kernels

Databricks Mountain View, California; San Francisco, California $166k–$225k/yr Published 6 months ago

NVIDIA CUDA DeepSpeed Megatron-LM PyTorch

Apply →

Forward Deployed Engineering Manager Remote

Labelbox San Francisco Bay Area $180k–$220k/yr Published 2 months ago

AWS Firecracker reinforcement learning Python GCP

Apply →

Staff Software Engineer, Model Serving

Databricks San Francisco, California $192k–$260k/yr Published 8 months ago

scheduling Routing AI/ML CPU distributed systems

Apply →

Software Engineer, ML Infrastructure

Cursor San Francisco Published 4 months ago

TypeScript Ray Golang Slurm Rust

Apply →

Staff Designated Support Engineer

Databricks San Francisco, California $141.7k–$250.8k/yr Published 1 year ago

AWS Snowflake Machine Learning Redshift Java

Apply →

Staff+ Infrastructure Engineer, Cluster Infrastructure Remote Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $320k–$4050k/yr Published 1 month ago

Terraform Istio AWS Argo Workflows Temporal

Apply →

Engineering Manager, Inference Routing and Performance Remote Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $405k–$485k/yr Published 2 months ago

continuous batching coordination planes prefill/decode disaggregation KV caching Trainium

Apply →

Staff + Senior Software Engineer, Inference Remote Visa

Join Anthropic's Inference team to design and maintain distributed systems that serve AI models to millions of users worldwide.

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $320k–$485k/yr Published 6 days ago

AWS Rust Python GCP Kubernetes

Flexible on stack

Apply →