SeniorHybrid$229K–$315K/yr

Matched in about the company · 3 passages

As a Senior Machine Learning Engineer, you will own the development of high-performance feature computation and online inference pipelines that power production machine learning systems at scale. You’ll build robust observability, monitoring, and automated debugging capabilities, while leveraging AI-assisted tools to investigate complex system behavior and…

PythonSparkAirflowPyTorch

Mid-levelHybrid$114K–$235K/yr est.

Matched in about the company · 2 passages

This role focuses on advancing the science and systems behind ML measurement, feature understanding, and causal inference at scale. The work spans areas such as production feature importance platforms, observational causal estimation in Pytorch, large-scale proxy metric development, and data-driven approaches to ML infrastructure efficiency. We're looking…

PythonSparkAirflowPyTorch

Staff+$220K–$292K/yr est.

Matched in about the company · 2 passages

As a Staff Engineer, you will own the technical roadmap for our ML platform. You will build the robust infrastructure, MLOps tooling, and systems architecture required to train, evaluate, host, and serve complex AI models (including LLMs, computer vision, and RL agents) in both cloud environments and air-gapped, offline tactical edge networks. You will be a…

PythonGoC++AWSKubernetesDockerPyTorchLLMs

Staff+

Matched in responsibilities

Training, validating, evaluating, and deploying cutting-edge machine learning algorithms including classifiers, LLMs, and computer vision models; · Building agentic systems for automated prompting, red-teaming, research, and rapid experimentation; · Supporting projects with specialized knowledge of frontier model architectures and cutting-edge technology.

PythonJavaGoKotlinSQLNode.jsGCPKubernetes

Hybrid$347K–$445K/yr

Matched in responsibilities · 3 passages

We’re looking for an experienced Software Engineer to help build the machine learning infrastructure that powers OpenAI’s monetization and ads systems. In this foundational role, you’ll design and develop the platform layer that enables teams to build, train, deploy, serve, monitor, and continuously improve machine learning models used across advertising…

On-site

Matched in responsibilities · 2 passages

As a Software Engineer on ML Platform at SpaceXAI, you'll build the infrastructure that turns real product usage into better models — and keeps research moving fast on large GPU fleets. ML Platform is organized into four teams. Depending on your background, you may join any of them: · Telemetry — Own the collection and serving path that turns real product…

KubernetesLinuxSpark

SeniorOn-site

Matched in responsibilities · 3 passages

You will lead a team of engineers building the infrastructure used to train, test, and evaluate our models. This is one of the few places at SpaceXAI where infrastructure and model behavior meet directly: when something breaks, it's rarely obvious whether it's a systems bug or the model doing exactly what it was trained to do, and your team has to be good…

Mid-levelRemote$160K–$220K/yr

Matched in details · 2 passages

Are excited by the challenge of operationalizing cutting-edge AI models at production scale · Believe that great infrastructure is what turns research breakthroughs into customer value · Enjoy designing systems that are automated, reliable, and self-healing · Want to work on problems where minutes of latency reduction or percentage points of accuracy matter…

PythonKubernetesDockerTerraform

Staff+$251K–$310K/yr est.

Matched in responsibilities · 3 passages

Design, build, and optimize realistic simulation environments and business logic running on TPUs using JAX and TensorFlow. Implement and optimize large-scale model and data parallelism strategies for training and running foundation models on TPU hardware. · Collaborate closely with modeling teams to integrate foundation models into simulation pipelines. ·…

PyTorchTensorFlowJAX

SeniorHybrid

Matched in responsibilities · 4 passages

The Data Intelligence & Analytics organization builds the core data platform and internal products that power decision-making across the company. We design and operate large-scale data systems, own the company’s data lake, ingestion infrastructure, and platform tooling, and develop end-to-end applications that transform complex datasets into fast, reliable,…

PythonTypeScriptJavaScriptReactFastAPIAWSGCPAzure

Hybrid$380K–$500K/yr

Matched in responsibilities · 2 passages

Design, build, and maintain data infrastructure systems such as distributed compute, data orchestration, distributed storage, streaming infrastructure, machine learning infrastructure while ensuring scalability, reliability, and security. · Ensure our data platform can scale by orders of magnitude while remaining reliable and efficient. · Partner with…

SeniorRemote$191K–$225K/yr est.

Matched in responsibilities · 2 passages

Own the design and reliability of ML inference infrastructure serving both predictive models and LLMs, maintaining high availability and low latency at scale. · Build and optimize low-latency streaming pipelines that deliver fresh, high-quality feature data to production ML models. · Drive improvements to distributed training infrastructure, enabling ML…

LLMs

SeniorRemote$217K–$303K/yr est.

Matched in responsibilities · 3 passages

Design and build large-scale offline ML experimentation platforms that enable reproducible research, model development, evaluation, and promotion workflows. · Develop production-grade training orchestration frameworks supporting distributed training, hyperparameter optimization, model evaluation, and automated retraining. · Build infrastructure for…

SparkAirflow

Staff+Remote$293K–$410K/yr est.

Matched in requirements

You have 8+ years of experience in infrastructure, distributed systems, ML platforms, data platforms, or large-scale backend systems. · You have 4+ years building or operating production ML infrastructure, feature platforms, training data systems, experimentation systems, or large-scale data pipelines. · You have led broad, ambiguous, multi-team platform…

KubernetesKafkaSparkAirflowSnowflake

Staff+Remote$244K–$305K/yr est.

Matched in details · 2 passages

12+ years of industry experience in applied ML/AI, inclusive MS or PhD in relevant fields · Strong programming (Scala / Python / Java / C++ or equivalent) and data engineering skills · Deep understanding of ML/AI best practices (e.g. training/serving skew minimization, A/B test, feature engineering, feature/model selection), algorithms (e.g. neural…

PythonJavaC++ScalaKubernetesKafkaAirflowPyTorch

InternOn-site

Matched in details

Agentic systems: Agentic Orchestration, Deep Research, Information Retrieval, Semantic Search, LLM code generation, LLM tool utilization, Textual RAG systems · Core ML/AI: LangGraph, Transformers, HuggingFace, LightGBM, PyTorch, SKLearn, XGBoost · Data Exploration & Visualization: Jupyter, Matplotlib, Pandas, Weights & Biases, Langfuse · Data Management &…

PythonFastAPIAWSDockerSparkAirflowPostgreSQLPyTorch

Hybrid$177K–$270K/yr

Matched in requirements

2+ years of ML infrastructure experience. · Experience with training frameworks like PyTorch, DeepSpeed, JAX, Ray, etc. · Experience working with cloud providers like AWS. · Bonus Qualifications: • Experience with building large-scale, cost-efficient distributed model training and ML compute infrastructure. · Experience with building model lifecycle…

AWSPyTorchJAX

Staff+$292K–$365K/yr est.

Matched in requirements

PhD in AI, machine learning, data science, or related technical fields. · Publications at peer-reviewed AI conferences (e.g., NeurIPS, CVPR, ICML, ICLR, ICCV, and ACL). · Customer Support Systems: Experience with AI technologies in customer support applications. · Agile Practice for AI production: Experience with the entire AI product development lifecycle…

PythonPyTorchLLMs

Matched in requirements · 3 passages

Have strong software engineering fundamentals and a track record of building performant, reliable systems · Have worked on ML infrastructure, distributed systems, or research tooling · Care about enabling other people's work and find leverage through platforms rather than individual experiments · Are comfortable operating across the stack, from low-level…

PyTorchJAX

Hybrid$500K–$850K/yr est.

Matched in requirements · 3 passages

Have strong software engineering fundamentals and a track record of building performant, reliable systems · Have worked on ML infrastructure, distributed systems, or research tooling · Care about enabling other people's work and find leverage through platforms rather than individual experiments · Are comfortable operating across the stack, from low-level…

PyTorchJAX