ML Infra Engineer: Scale Ray + PyTorch for Multimodal AI
Orbifold AI
Orbifold AI is seeking a Machine Learning Engineer in Palo Alto, California, to build and optimize the infrastructure for multimodal data processing. You will play a key role in scaling our Ray and PyTorch-based systems utilized in frontier robotics. The ideal candidate will have extensive experience with distributed systems, strong proficiency in Python, and proven knowledge of machine learning frameworks, particularly PyTorch. You will directly influence the efficiency and performance of our data processing and model training pipelines. #J-18808-Ljbffr Orbifold AI
Vacancy posted 1 day ago
Similar jobs that could be interesting for youBased on the ML Infra Engineer: Scale Ray + PyTorch for Multimodal AI in Palo Alto, CA vacancy
- 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...Suggested
- Rhoda AI is building next-generation generalist robots and is seeking a Staff / Principal ML Training Systems Engineer to own training systems performance end-to-end. You will optimize large-scale multimodal training, define parallelism strategies, and drive efficiency...Suggested
- Wanderboat AI is hiring ML engineers to build the core intelligence behind search, recommendations, and multimodal retrieval. You will apply machine learning, deep learning, and generative AI to real product challenges at scale, shaping retrieval, ranking, and content...Suggested
- Rhoda AI in Mountain View is seeking a Staff / Principal ML Training Systems Engineer to lead the performance of large-scale multimodal training systems. This role involves improving training efficiency... ...with modern ML platforms like PyTorch. You will contribute...Suggested
- Scale our Ray + PyTorch infrastructure for the multimodal video, training, and RL pipelines that power frontier robotics... ...3+ yrs distributed systems / ML infra. About Orbifold AI Orbifold AI is building the... ...hiring a Machine Learning Engineer to scale and optimize the ML...Suggested
- Rhoda AI is seeking a Staff / Principal ML Training Systems Engineer in Mountain View to enhance the training systems performance... .... This role focuses on large-scale multimodal training, driving efficiency,... ..., significant experience with PyTorch, and strong systems intuition....
- # Machine Learning Engineer, Multimodal AIGoogle DeepMind## Job Description### About Google... ...technological challenges through AI.### Responsibilities- Build... ...architectures.- Proficiency in JAX or PyTorch and experience with large-scale data pipelines.- M.S. or Ph.D. in...Flexible hours
$251k - $310k
...for researchers and software engineers who are passionate about... ...technical leadership on large-scale ML model architectures,... ...tooling and frameworks e.g. PyTorch, Jax, Tensorflow, Ray, or similar. The candidate... ...metrics for evaluating complex AI systems. Track record of technical...Full time$230k - $265k
...deploy cutting-edge AI technology to... ...for ML and work alongside... ...scientists and engineers. As a Senior Machine... ...learning in order to scale and optimize... ...models using PyTorch and/or JAX,... ...experience in ASR, TTS, multimodal, or modern LLM/... ...with product, infra, research, and...Permanent employment- ...and hyper-connected world.Role OverviewAs our Staff Software Engineer, ML infra Engineer for Search & Discovery organization, you will be in... ...logging, processing, and ML model training pipelinesDevelop and scale data infrastructure that powers batch and real-time data...Temporary work
$189k - $300k
...of transportation on a global scale. The Data Scaling team owns... ...directly works on and delivers ML models to the product that... ...collaborative, high-impact team of AI/ML engineers, data scientists and... ...and timelines. Proficiency in PyTorch and Python. Experience building...Full timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours$251k - $310k
...ultimate cognitive engine for autonomous... ...the use of large multimodal foundation models... ...autolabeling at a massive scale. This closed-loop... ...Cross-Functional AI Strategy: Act as... ...visionary across ML Infra, Perception, Behavior... ...using Python, PyTorch, or JAX, with a track...Full timeRemote workNeura Market
Mountain View, CA4 days ago- ...applications for a Sr. Machine Learning Engineer, Siri Speech to advance multilingual, multimodal LLMs and speech understanding... .... You will design end-to-end ML systems, train models, and... ...to deliver private, user-centric AI experiences at scale. #J-18808-Ljbffr Apple
- ...: As a Senior Machine Learning Engineer, you will be responsible for building... ...the power of Generative AI to our customers. You will work... ...Generative AI models at significant scale. Investigate, prototype and... ...Experience with building and evolving ML Training and Inferencing...Local area
$140k - $230k
...mobility; and Cloud & AI, the digital... ...petabytes of multimodal driving data, solving... ...learning (ML) training and evaluation... ...Software Engineer or Machine Learning... ...accelerate and scale up our overall... ...), Python, and PyTorch/Tensorflow. ~2... ..., Flyte, Flink, Ray, or similar ML pipelines...Temporary workWork at officeFlexible hours3 days per weekWoven by Toyota
Palo Alto, CA11 days ago- ...Factory is building an AI training and model... ...infrastructure that makes large-scale training and RL... ...pipelines, and production engineering. Your responsibilities... ...-LM, DeepSpeed, PyTorch FSDP/DTensor, Ray, verl, slime, AReaL, OpenRLHF... ...training, large-scale ML systems, or GPU...
Nebius B.V.
Palo Alto, CA5 days ago $180k
...leading technology company is seeking expert engineers for a role focused on multimodal mid-training data. Candidates will design algorithms... ...and optimize data mixtures. Expertise in ML and familiarity with large model scaling are essential. The position offers a...Relocation- Strativ Group in Palo Alto is building AI infrastructure for the next generation of agentic AI. You will design and optimize production infrastructure for large-scale AI workloads, focusing on TPU and GPU inference, distributed serving, and runtime performance. As part...
$250k - $320k
Staff Infrastructure Engineer We are partnered with a Stealth AI Lab (backed by top-... ...media, simulation, and multimodal interaction,... ...content. As a Staff Infra Engineer, you will work... ...architect, build, and scale the compute... ...experience in Software / ML Infrastructure Engineering...Full timeImmediate start- ...next generation of AI builders, and... ...scientists, and engineers, tackling the most... ...a distributed ML infrastructure engineer... ...help extend and scale our training... .... • Infra Engineering – Write... ...for ML infra in PyTorch or JAX; ensure reliability... ..., Kubernetes, Ray) and debugging...Flexible hours
Institute of Foundation Models
Sunnyvale, CA26 days ago $180k
...s mission is to create AI systems that can accurately... ..., and focused on engineering excellence. This organization... ...ABOUT THE ROLE: As an ML Infrastructure Engineer... ..., building, and scaling GPU compute infrastructure... ...frameworks such as JAX or PyTorch Low-level...Temporary workWork experience placementSpaceXAI
Palo Alto, CA17 days ago$166k - $244k
...working to build the AI-powered electric grid... ...energy, AI, software engineering, and products to build... ...matters at global scale. Learn more about our... ...Engineers across diverse ML domains spanning multimodal machine learning,... ...with ML frameworks like PyTorch or TensorFlow....Full timeFlexible hours$155.42k - $395.9k
...DescriptionAbout the Team:The ML Compute Platform is part of the AI Compute Platform... ...looking for a Senior Software Engineer to join our team and help us scale our platform for... ...training frameworks like PyTorch, TorchX Experience with Ray framework Leadership/active...Full timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hours$170.6k - $261.3k
...join us. About the team: The AV ML Infra team at GM builds end-to-end... ...to meet the unique demands of AI and ML innovation, supporting... ...enhance the productivity of ML engineers, and drive the adoption of cutting... ...performance by running large-scale simulation workloads and...Full timeLocal areaWork from homeFlexible hours- A leading AI infrastructure company in California seeks a Member of Technical Staff — Training to design and optimize large-scale distributed training systems for frontier AI models. Candidates... ...have 5+ years of experience in ML systems and be proficient in Python...
$262k - $361k
...working to build the AI-powered electric... ..., AI, software, engineering, and product to build... ...matters at global scale. Learn more about... ...of production ML/AI systems. You will... ...architecture for Tapestry’s multimodal intelligence... ...such as PyTorch, JAX, or TensorFlow...Full timeRemote workFlexible hours$153.2k - $234.1k
...of transportation on a global scale. Role:Are you passionate... ...autonomous driving? Join the Embodied AI team at General Motors. Our... ...world scenarios. As a Senior ML Infra Engineer, you will work on the core... ...learning frameworks such as PyTorch or TensorFlow. Experience with...Full timeLocal areaRemote workWork from homeRelocation packageFlexible hours$124k - $250k
...in the LifeAs a member of our software engineering infra team, you'll solve technical challenges,... ...MakeDesign, develop, and maintain large-scale distributed systemsCollaborate with various... ..., including artificial intelligence (AI), to help identify and evaluate candidates...$230k - $260k
...intersection of creativity and AI with real impact. Join us to... ...a Principal Machine Learning Engineer, you will operate at the... ...will lead the design of large-scale ML systems and shared platforms... ...powering agentic orchestration and multimodal content generation Define and...Work at officeImmediate start3 days per week$229k - $343k
....Snap’s Generative ML Platform team builds cutting-edge AI technologies that power... ...worldwide. From multimodal LLMs and video... ...a Machine Learning Engineer to join Snap Inc!What... ...such as TensorFlow, PyTorch, JAX, MLX, scikit-learn... ...training large-scale diffusion models for...Full timeLive inWork at officeLocal areaWorldwide
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to ML Infra Engineer: Scale Ray + PyTorch for Multimodal AI. 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
- artificial intelligence - machine learning intern Palo Alto, CA
- internship machine learning Palo Alto, CA
- machine learning researcher Palo Alto, CA
- machine learning Palo Alto, CA