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
Vacancy posted 3 days 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 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
- ...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, CA5 days ago - ...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...PipelineFlexible hours
Institute of Foundation Models
Sunnyvale, CA26 days ago $180k
...highly motivated, and focused on engineering excellence. This organization... .... ABOUT THE ROLE: As an ML Infrastructure Engineer, you... ...Developing data pipelines and integrating large-scale data... ...ML frameworks such as JAX or PyTorch Low-level understanding of...PipelineTemporary workWork experience placementSpaceXAI
Palo Alto, CA17 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...PipelinePermanent employment$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$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$180k
...highly motivated, and focused on engineering excellence. This organization... ...Leverage all of SpaceXAI's infra and AI stacks to dramatically... ...user experience Write data pipelines and training jobs that continuously... ...frameworks such as JAX or PyTorch Exceptional candidates may be...PipelineTemporary workPantera Capital
Palo Alto, CA1 day 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...PipelineFull timeLocal areaRemote workWork from homeRelocation packageFlexible hours$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 workLatent Logic
Mountain View, CA5 days ago- ...and deploy production‑grade ML systems with end‑to‑end ownership... ...experience in ML engineering. Strong programming skills in... ...experience with ML frameworks such as PyTorch or TensorFlow. Familiarity with... ...Strong understanding of data pipeline design, real‑time inference,...PipelineFull time
Catalyst Labs, LLC
Palo Alto, CA5 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...PipelineFull timeRemote workNeura Market
Mountain View, CA4 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- ...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
- ...AI, headquartered in Mountain View, CA, is seeking an AI engineer to advance its ASR pipeline and deploy deep learning models for speech recognition.... ...setting. The role requires 2+ years in DL for speech, a CS degree, and strong Python/PyTorch skills. #J-18808-Ljbffr Level AIPipeline
$140k - $230k
...efficient machine learning (ML) training and evaluation pipelines, and designing novel... ...looking for a skilled Software Engineer or Machine Learning... ...or similar), Python, and PyTorch/Tensorflow. ~2+ years of... ...Spark, Airflow, Flyte, Flink, Ray, or similar ML pipelines technologies...PipelineTemporary workWork at officeFlexible hours3 days per weekWoven by Toyota
Palo Alto, CA11 days 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
$117k - $152k
...Unity Vector builds an offline ML platform that powers insight,... ..., machine learning pipelines, and business operations. As... ...looking for a Machine Learning Engineer to join our Offline Infrastructure... ...distributed training workflows (e.g., PyTorch, Ray) Work with workflow...PipelineWork at officeRemote workWorldwideRelocation package- ...Technologies is seeking a Machine Learning Data Engineer to build and operate large-scale data systems powering AI training pipelines. The role emphasizes ingestion,... ...JVM language skills, and expertise with Spark, Ray, or Beam and petabyte-scale storage. This is...PipelineRemote job
$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$307k - $422k
...and develop a diverse team of ML/AI experts, as well as guide... ...learning).Provide research and engineering leadership on X early... ...strategy and contribute to ML pipeline development and system evaluation... ...frameworks (e.g., TensorFlow, PyTorch, JAX, HuggingFace, LangChain)...PipelineFull time$228k - $285k
...generations. Role Summary As a Staff Software Engineer for ML Optimization and Hardware Acceleration,... ...: Design and build automated pipelines for regular model profiling across diverse... ...Proficiency in Python and deep knowledge of PyTorch or TensorFlow. Hands‑on experience with...PipelineFull timeContract workLocal area$207k - $300k
...infrastructure that powers our AI/ML applications. You will operate at the... ..., including creative data engines, generation and rendering pipelines, and agentic orchestration frameworks... ...learning frameworks (e.g., TensorFlow, PyTorch) and cloud/distributed data technologies...Pipeline- ...Technologies seeks a Machine Learning Engineer to join the Offline... ...focusing on large-scale data pipelines and distributed model training... ...datasets, support multi-stage ML workflows, and collaborate with... ...production, offering exposure to PyTorch, Spark, and orchestration tools...Pipeline
$160k - $200k
...growing teams. As a Senior ML Infrastructure Engineer at Plus, you will design... ...phases. You will build robust pipelines for managing model... ...deep learning frameworks like PyTorch or TensorFlow. If you are eager... ...computing frameworks like Ray Experience working with...PipelinePlusAI
Santa Clara, CA26 days ago$300k - $400k
...scheduling for GB series GPUs across Ray, Slurm, and Kubernetes,... ...without blocking the training pipeline Engage directly with the SGLang... ...and benchmarking distributed ML systems to identify and... ...world’s best — the scientists, engineers, and problem-solvers who don’t...PipelineVisa sponsorshipFlexible hoursShift work
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to ML Infra Engineer — Ray + PyTorch Pipelines. Be the first to apply!
Related searches
- ai ml engineer Palo Alto, CA
- senior ml engineer Palo Alto, CA
- machine learning ai engineer Palo Alto, CA
- computer vision machine learning engineer Palo Alto, CA
- machine learning engineer Palo Alto, CA
- machine learning software engineer Palo Alto, CA
- pipeline Palo Alto, CA
- labor union pipeline Palo Alto, CA
- pipeline tech Palo Alto, CA
- pipeline surveying Palo Alto, CA