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- 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 AIPipeline
- ...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...PipelineTemporary work
- 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
- ...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
- ...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
- ...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
- ...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...PipelineFull time
- ...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,...PipelineFull time
$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,...PipelineFull timeImmediate startFlexible hours$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...PipelineFull timeLocal areaWork from homeFlexible hours$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...PipelineFull timeRemote work$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...PipelinePermanent employment$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...PipelineFull timeFlexible hours$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...PipelineFull timeTemporary workImmediate startRemote work$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$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...PipelineFull timeRemote work$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:...PipelineFull timeRemote work$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)...PipelineFull timeTemporary workRemote work$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$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$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...PipelineFull timeLocal areaRemote workWork from homeRelocation packageFlexible hours- ...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
$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...PipelineFull timeRemote work$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...PipelineFull timeLocal areaRelocationVisa sponsorship$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...PipelineFull timeRemote work- ...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...PipelineWork from homeRelocation packageFlexible hours
- ...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...PipelineWork at officeRemote workFlexible hours
$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...PipelineFull timeImmediate start- 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
- ...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-LjbffrPipeline
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to ML Infra Engineer — Ray + PyTorch Pipelines. Be the first to apply!
- ai ml engineer Palo Alto, CA
- senior ml engineer Palo Alto, CA
- computer vision machine learning engineer Palo Alto, CA
- machine learning engineer Palo Alto, CA
- machine learning ai engineer Palo Alto, CA
- machine learning software engineer Palo Alto, CA
- pipeline Palo Alto, CA
- pipeline tech Palo Alto, CA
- pipeline surveying Palo Alto, CA
- labor union pipeline Palo Alto, CA




