Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

ML Infra Engineer — Ray + PyTorch Pipelines

Orbifold AI

Orbifold AI in Palo Alto, CA is seeking a Machine Learning Engineer to scale and optimize the ML infrastructure behind our multimodal data pipelines. You will work with PyTorch and Ray to convert raw video, image, and sensor data into training, evaluation, and RL signals for frontier robotics teams.

This role emphasizes building scalable, fault-tolerant systems and delivering end-to-end infrastructure for researchers and product engineers in a fast-paced startup environment.

#J-18808-Ljbffr
Vacancy posted 14 hours ago
Similar jobs that could be interesting for youBased on the ML Infra Engineer — Ray + PyTorch Pipelines in Palo Alto, CA vacancy
  • Orbifold AI is seeking a Machine Learning Engineer in Palo Alto, California, to build and...  ...will play a key role in scaling our Ray and PyTorch-based systems utilized in frontier robotics...  ...of our data processing and model training pipelines. #J-18808-Ljbffr Orbifold AI
    Pipeline

    Orbifold AI

    Palo Alto, CA
    4 days ago
  •  ...and hyper-connected world.Role OverviewAs our Staff Software Engineer, ML infra Engineer for Search & Discovery organization, you will be in...  ...with deploying highly robust and scalable data pipelines processing petabytes of dataExperience in managing mission critical... 
    Pipeline
    Temporary work

    Coupang

    Mountain View, CA
    2 days ago
  • Scale our Ray + PyTorch infrastructure for the multimodal video, training, and RL pipelines that power frontier robotics and world model teams...  ...+ yrs distributed systems / ML infra. About Orbifold AI Orbifold...  ...hiring a Machine Learning Engineer to scale and optimize the ML... 
    Pipeline

    Orbifold AI

    Palo Alto, CA
    4 days ago
  •  ...technology company in Mountain View is seeking an experienced professional to enhance ML infrastructure for training workloads. Responsibilities include designing distributed input data pipelines and collaborating with research teams to optimize systems. Candidates should... 
    Pipeline

    Waymo

    Mountain View, CA
    14 hours ago
  •  ...Description: We are looking for a senior ML infrastructure engineer to build and evolve the systems that...  .... Practical experience with PyTorch and ML training workflows, including...  ...performance computing ML training pipelines, job orchestration, compute scheduling... 
    Pipeline

    Maxinsights

    Santa Clara, CA
    28 days ago
  •  ...performance, model training frameworks, RL pipelines, and production engineering. Your responsibilities Build and...  ...such as Megatron-LM, DeepSpeed, PyTorch FSDP/DTensor, Ray, verl, slime, AReaL, OpenRLHF, or...  ...model training, large-scale ML systems, or GPU cluster workloads... 
    Pipeline

    Nebius B.V.

    Palo Alto, CA
    3 days ago
  •  ...Senior ML Infrastructure Engineer (PyTorch, Kubernetes, GPU Training) Short Job Description We are seeking a Senior ML Infrastructure Engineer...  ...GPU training platforms, optimize distributed training pipelines, and improve the developer experience for ML researchers... 
    Pipeline
    Full time

    Finoit Inc

    Redwood City, CA
    3 days ago
  •  ...the Role We’re looking for an Applied ML Engineer to design, evaluate, and scale recommendation...  ...generation, ranking, and re-ranking pipelines using offline and online signals. Develop...  ...open-source systems. Proficiency in PyTorch (or equivalent) and experience training,... 
    Pipeline
    Full time

    Darwin

    Palo Alto, CA
    9 hours ago
  • $175k - $275k

     ...us to power their data pipelines. With our headquarters...  ...datasets, or full-cycle data engineering, Abaka AI provides the...  ...machine learning or ML engineering, with a...  ...ML frameworks such as PyTorch, TensorFlow, or JAX, with...  ...tools such as Kubernetes, Ray, Airflow, or MLflow,... 
    Pipeline
    Full time
    Immediate start
    Flexible hours

    Abaka Ai

    Palo Alto, CA
    9 hours ago
  • $170.6k - $261.3k

     ...vehicles, join us. About the team: The AV ML Infra team at GM builds end-to-end ML...  ...experimentation, enhance the productivity of ML engineers, and drive the adoption of cutting-edge...  ...workloads and managing reliable ML inference pipelines. ML Compute: Streamlines and optimizes... 
    Pipeline
    Full time
    Local area
    Work from home
    Flexible hours

    General Motors

    Sunnyvale, CA
    1 day ago
  • $204k - $259k

     ...AV stack. We are an advanced ML and engineering team that leverages state-of...  ...closely with the ML Infra, Perception, Behavior, and AI...  ...Python and frameworks like PyTorch, JAX, or TensorFlow. ~1+ years...  ...large-scale data processing pipelines for ML training. ~ Proven... 
    Pipeline
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $196k - $221k

     ...AI team responsible for ML and work alongside...  ...veteran scientists and engineers. As a Machine Learning...  ...and speech models using PyTorch and/or JAX, making principled...  ...infrastructure, data pipelines, evaluation frameworks,...  ...end-to-end with product, infra, research, and data... 
    Pipeline
    Permanent employment

    Otter.ai

    Mountain View, CA
    2 days ago
  • $300k

     ...data scientists, and engineers, tackling the most...  ...for a distributed ML infrastructure engineer...  ...researchers.  • Infra Engineering –...  ...tests for ML infra in PyTorch or JAX; ensure reliability...  ...Slurm, Kubernetes, Ray) and debugging...  ...custom training pipelines at scale and modifying... 
    Pipeline
    Full time
    Flexible hours

    Institute Of Foundation Models

    Sunnyvale, CA
    9 hours ago
  • $195.2k - $262.2k

     ...building large in-house AI/ML infrastructure. Built by engineers, for engineers. From...  ...training frameworks, RL pipelines, and production engineering...  ...Strong Python and PyTorch engineering skills....  ...Inference Server, NVIDIA Dynamo, Ray Serve, KServe, or equivalent... 
    Pipeline
    Full time
    Temporary work
    Immediate start
    Remote work

    Nebius

    Palo Alto, CA
    1 day ago
  • $170k - $216k

     ...looking for researchers and software engineers who are passionate about developing machine...  ...distributed systems covering the ML lifecycle, supporting planet-scale...  ...learning tooling and frameworks e.g. PyTorch, Jax, Tensorflow, Ray, or similar. The candidate should understand... 
    Full time

    Waymo

    Mountain View, CA
    2 days ago
  • $251k - $310k

     ...build the ultimate cognitive engine for autonomous driving. We are...  ...(RLHF/RLAIF, DPO/GRPO/PPO) pipelines. Drastically improve the model...  ...principal technical visionary across ML Infra, Perception, Behavior, and AI...  ...fundamentals using Python, PyTorch, or JAX, with a track record... 
    Pipeline
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $204k - $259k

     ...collaborative group of machine learning (ML) engineers, software engineers, and ML research...  ...Architect and implement scalable and robust ML pipelines for training, evaluating, and deploying...  ...with ML frameworks such as Jax/Flax, PyTorch or Tensorflow. We prefer:... 
    Pipeline
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $238k - $302k

     ...Senior Staff Software Engineer. You will:...  ...scale data and evaluation pipelines You have: ~...  ...years of experience in ML engineering and applied...  ...We prefer: ML infra experience: training, evaluating...  ...an ML framework (e.g. Pytorch, JAX, Tensorflow)... 
    Pipeline
    Full time
    Temporary work
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $124k - $250k

     ...support of others.A Day in the LifeAs a member of our software engineering infra team, you'll solve technical challenges, including upgrading...  ...to our bidding ecosystem. You will be exposed to the whole pipeline of model delivery, including training, serving, and... 
    Pipeline

    AppLovin

    Palo Alto, CA
    2 days ago
  • $193.93k - $291.15k

     ...Nuro in Mountain View is seeking an experienced engineer to work on its Mapping team, focusing on building and improving HD map generation and release pipelines. You will develop scalable workflows and manage APIs while engaging with other mapping teams. Required qualifications... 
    Pipeline

    Nuro

    Mountain View, CA
    14 hours ago
  • $153.2k - $234.1k

     ...across real-world scenarios. As a Senior ML Infra Engineer, you will work on the core systems that...  ...next. You will develop model training pipelines that are performant, easy to use, and...  ...with deep learning frameworks such as PyTorch or TensorFlow. Experience with performance... 
    Pipeline
    Full time
    Local area
    Remote work
    Work from home
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    9 hours ago
  •  ...Job Title: ML Engineer What You Will Own End‑to‑End ML Lifecycle across real products: data...  ...Production‑grade ML systems built with PyTorch or TensorFlow, focusing on latency, reliability...  ..., CI/CD, automated testing, deployment pipelines, serving layers, monitoring and A/B... 
    Pipeline

    MetAntz

    Palo Alto, CA
    4 days ago
  • $175k - $215k

     ...Monitoring and event response engine to support expansion to...  ...and standardize the ML infrastructure for...  ...automated training/inference pipelines, low-latency spatial...  ...such as TensorFlow, PyTorch or Keras.We prefer:MS in...  ...on top of a mature ML infra.Experience working at another... 
    Pipeline
    Full time
    Remote work

    Latent Logic

    Mountain View, CA
    3 days ago
  • $220k - $300k

     ...planning, and decision-making pipelines that make autonomous systems...  ...AI researchers and hardware engineers to solve hard, interdisciplinary...  ...Do Develop and optimize ML models for perception, motion...  ...~ Hands-on experience with PyTorch and/or TensorFlow for model... 
    Pipeline
    Full time
    Local area
    Relocation
    Visa sponsorship

    Clera

    Mountain View, CA
    1 day ago
  • $175k - $215k

     ...Build and operate the petabyte-scale data systems and ML pipelines at the heart of Waymo's foundation model development Shepherd...  ...Familiarity with one of the modern deep learning frameworks (e.g. Pytorch, JAX, Tensorflow) Experience building or maintaining large... 
    Pipeline
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  •  ...As a Machine Learning Engineer in AI Core, Data Intelligence...  ...and feature pipelines that feed our state-of-...  ...multiple stages of the ML lifecycle: ingesting and...  ...computing frameworks like Ray. Tune and optimize machine...  ...Core ML, Platform, and Infra teams to ensure... 
    Pipeline
    Work from home
    Relocation package
    Flexible hours

    Socket

    Palo Alto, CA
    2 days ago
  •  ...with Moveworks’ Reasoning Engine and natural language...  ...help build cutting edge ML infrastructure for...  ...learning systems. The ML infra team covers a variety of...  ...training and inference pipeline for large language models...  ...framework such as Pytorch or Huggingface or LLM serving... 
    Pipeline
    Work at office
    Remote work
    Flexible hours

    Moveworks

    Mountain View, CA
    3 days ago
  • $250k - $320k

     ...Staff Infrastructure Engineer We are partnered with a Stealth AI Lab...  ...immersive content. As a Staff Infra Engineer, you will work directly...  ...clusters, and data processing pipelines that drive real‑time...  ...years of experience in Software / ML Infrastructure Engineering. Deep... 
    Pipeline
    Full time
    Immediate start

    Strativ Group

    Menlo Park, CA
    3 days ago
  • Yahoo Mail is seeking a Machine Learning Engineer II to design and deploy high-throughput AI/ML and NLP models at scale, powering personalized...  ...with Python or Java, experience with PyTorch/TensorFlow, and modern ML pipelines on Spark/Hadoop. #J-18808-Ljbffr Yahoo Holdings... 
    Pipeline

    Yahoo Holdings Inc. in

    Mountain View, CA
    2 days ago
  •  ...Apple is seeking an ML Infrastructure Engineer focused on ML user experience APIs and integration for on-device AI. You will develop new model...  ...these APIs into internal and external model repositories, optimize pipelines from authoring to runtime, and #J-18808-Ljbffr
    Pipeline

    Socket.dev

    Cupertino, CA
    14 hours ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to ML Infra Engineer — Ray + PyTorch Pipelines. Be the first to apply!