"ml training infrastructure" Jobs

Showing 20 of 72 results

Staff Backend Software Engineer

Databricks New York $190k–$261.3k/yr Published 2 months ago

Ray system observability Sagemaker Azure ML SGLang

Apply →

Software Engineer, Data Platform

Ramp New York, NY (HQ) Published 3 months ago

Terraform Datadog AWS Snowflake Redshift

Apply →

Machine Learning Systems Engineer, RL Engineering Remote Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $500k–$850k/yr Published 8 months ago

distributed systems reinforcement learning LLM Python RLHF

Apply →

Senior Staff+ Software Engineer, Node Infra Remote Visa

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

Terraform Trainium Go RDMA EFA

Apply →

Product Manager II - Model Lab Remote

Datadog New York, New York, USA $155k–$190k/yr Published 9 months ago

JAX AI/ML PyTorch TensorFlow

Apply →

Machine Learning Systems Engineer, Research Tools Remote Visa

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

Large Language Models Machine Learning distributed systems WordPiece Transformer-based architectures

Apply →

Staff + Senior Software Engineer, Inference Deployment Remote Visa

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

Trainium Rust TPU Python GPU

Apply →

Technical Program Manager, Infrastructure Remote Visa

Anthropic San Francisco, CA | New York City, NY | Seattle, WA $290k–$365k/yr Published 4 months ago

AWS CI/CD Trainium AI tools TPU

Apply →

Manager II, Engineering - AI Platform Training, Serving and Storage (NorAm) Remote

Datadog New York, New York, USA $234k–$300k/yr Published 4 months ago

Ray AI/ML Open Source Tooling AnyScale

Apply →

Software Engineer, Machine Learning Remote

Figma San Francisco, CA • New York, NY • United States $153k–$376k/yr Published 1 year ago

Spark MLlib scikit-learn XGBoost PyTorch TensorFlow

Apply →

Research Engineer, Machine Learning (RL Velocity) Remote Visa

Anthropic Remote-Friendly (Travel-Required) | San Francisco, CA | New York City, NY $500k–$850k/yr Published 1 month ago

JAX AI/ML PyTorch

Apply →

Full-Stack Software Engineer, Reinforcement Learning Remote Visa

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

TypeScript AWS React Python GCP

Apply →

Sr. Manager, Field Engineering - Lakebase Remote

Lead a specialized team of pre-sales Solutions Architects focused on driving the growth of Databricks' Lakebase product offering.

Databricks Atlanta, Georgia; Chicago, Illinois; Dallas, Texas; New York City, New York; San Francisco, California; United States $218.4k–$300.3k/yr Published 1 week ago

AI solutions OLTP Data Platforms cloud infrastructure Data Warehousing

Apply →

Staff Software Engineer - Agent Quality

Databricks New York City, New York $190k–$270k/yr Published 1 month ago

C++ Ray AI/ML Java Go

Apply →

Staff Software Engineer - AI Research Infrastructure

Databricks New York City, New York; San Francisco, California $190k–$270k/yr Published 1 month ago

C++ Ray AI/ML Java Go

Apply →

Staff Software Engineer - AI Research Infrastructure

Develop and run the research stack that powers Databricks AI Research, enabling rapid large-scale experiments.

Databricks New York City, New York; San Francisco, California $199k–$270k/yr Published 1 month ago

C++ Ray Java Go Slurm

Flexible on stack

Apply →

Staff Database Engineer

Databricks Mountain View, California; New York City, New York; San Francisco, California $192k–$260k/yr Published 2 months ago

OLTP PostgreSQL Data Intelligence Platform

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 GenAI Engineer - Application Performance Monitoring (APM) Remote

Join Datadog as a Staff GenAI Engineer to lead machine learning projects within the Application Performance Monitoring team.

Datadog New York, New York, USA $234k–$234k/yr Published 1 week ago

Machine Learning APM ML GenAI

AI-first team

Apply →

Senior Staff GenAI Engineer - Application Performance Monitoring (APM) Remote

Datadog New York, New York, USA $272k–$340k/yr Published 1 year ago

ML GenAI AI

Apply →