PyTorch Jobs

San Francisco, CA, US; Remote, US8d ago

Python Snowflake Machine learning TensorFlow PyTorch Scikit-learn Spark A/B testing Documentation Ownership+1

Master’s, or PhD in Machine Learning, Computer Science, Statistics, Applied Mathematics, Engineering, or related quantitative fields; or equivalent applied…

Sr. Staff Machine Learning Engineer, Homefeed

Python Java Machine learning Deep learning TensorFlow PyTorch Spark

Languages: Python, Java Machine Learning: PyTorch, Tensorflow Big data processing: Spark, Hive, MapReduce

Senior Machine Learning Engineer II, NLU & Agentic AI

Moveworks

Mountain View, CA8d ago

Python Go Machine learning Deep learning PyTorch NLP Security Ownership Transformers Few-shot prompting+5

We are looking for an experienced software engineer with machine learning expertise to join us in expanding Moveworks NLU (natural language understanding) and…

Senior Staff Machine Learning Engineer, (ML Underwriting)

Affirm

Python REST Machine learning Deep learning PyTorch Spark A/B testing Mentoring Documentation MLflow+4

You have 10+ years of experience researching, designing, deploying, and operating large-scale, real-time machine learning systems, with a proven record of…

Senior Applied AI/ML Scientist - Search Ads

Faire

San Francisco, CA8d ago

Machine learning Deep learning PyTorch LLM NLP Ownership FAISS Pinecone Large language models Embeddings

4+ years of experience building and shipping ML systems in production, with meaningful experience in search, recommendation, or ads ranking/retrieval.…

Senior Staff Data Scientist

FourKites

Chennai or Remote, India8d ago

Toronto, Ontario, Canada8d ago

Python SQL TimescaleDB AWS GCP Azure Machine learning Deep learning TensorFlow PyTorch+9

Education & Experience Master's degree in Data Science, Statistics, Computer Science, Mathematics, Operations Research, Industrial Engineering, or related…

Senior Director/VP - Head of AI Engineering

CommerceIQ

Director

Python Java C++Scala Machine learning Deep learning TensorFlow PyTorch A/B testing Mentoring+2

5+ years of experience in backend software engineering, with 2+ years leading AI/ML teams.Hands-on experience training AI/ML models, including foundation…

2026 Summer Intern, PhD, ML Training Infra and Efficiency

Waymo

Intern

Python C++Machine learning Deep learning TensorFlow PyTorch Distributed training

Python or C++ experience Scale model training to multiple accelerators Reinforcement learning

GenAI Engineer

Waymo

Mountain View CA USA8d ago

Python Machine learning TensorFlow PyTorch LangChain AI agents

Master’s in Computer Science, Artificial Intelligence, or related field Experience with Generative AI (LLMs) and its application in enterprise business…

Machine Learning Engineer, Data & Systems

Waymo

Mountain View, California, United States, San Francisco, California, United States8d ago

Python C++Machine learning Deep learning PyTorch Computer vision Ownership

MS or PhD Degree in Machine Learning, Robotics, Computer Science or a similar discipline Publications at top-tier conferences like CVPR, ICCV, ECCV, ICLR,…

Machine Learning Engineer, Mapping

Waymo

C++Machine learning Deep learning TensorFlow PyTorch Computer vision Ownership Few-shot prompting

Design, train, and deploy machine learning models to automate the creation of Waymo's HD maps, unlocking scale for the Waymo Driver. Apply and advance…

Machine Learning Engineer, Perception Modeling

Waymo

Mountain View, CA, USA; San Francisco, CA, USA8d ago

Python C++Machine learning Deep learning PyTorch Computer vision Ownership Model evaluation

MS or PhD Degree in Machine Learning, Robotics, Computer Science or a similar discipline Publications at top-tier conferences like CVPR, ICCV, ECCV, ICLR,…

Machine Learning Engineer, Runtime & Optimization

Waymo

Mountain View, California, USA8d ago

Python C++Machine learning Deep learning PyTorch LLM Mentoring Large language models CUDA

PhD in CS, EE, Deep Learning or a related field. Familiarity with the HW architecture of ML hardware accelerators (e.g., GPU/TPU). Deep knowledge of model…

Machine Learning Engineer, Simulation Realism

Waymo

Python C++Machine learning TensorFlow PyTorch

Experience working with, creating and/or developing ML models for robotics and/or self driving vehicles Strong C++ in production setting, with experience in…

Senior Machine Learning Engineer, Perception, Semantics

Waymo

Mountain View, California, USA; Kirkland, WA, USA8d ago

Machine learning Deep learning TensorFlow PyTorch Computer vision Ownership Few-shot prompting

PhD degree in Computer Science or a similar discipline, or an equivalent amount of deep learning experience. Experience with multi-modal perception systems…

Senior Machine Learning Infrastructure Engineer, Simulation

Waymo

Mountain View, CA8d ago

Machine learning TensorFlow PyTorch Mentoring Ownership Transformers Distributed training DeepSpeed

10+ years of professional software engineering experience, with at least 5 years in machine learning infrastructure such as developing, designing, scaling,…

Senior Research Scientist, Foundation Model (LLM/VLM)

Waymo

Mountain View, CA, USA; San Francisco, CA, USA; New York City, NY, USA8d ago

Python Machine learning Deep learning TensorFlow PyTorch Documentation Embeddings Llama

Experience with: Generative modeling LLMs or VLMs (e.g., Gemini, Llama, GPT)

Senior Staff TLM, Perception, Object Detection

Waymo

Mountain View, CA, USA; San Francisco, CA, USA8d ago

Python C++Machine learning Deep learning PyTorch Computer vision Ownership

Own object detection quality, work with product and safety partners on short- and long-term improvement goals in support of Waymo’s scaling goals Develop and…

Staff Machine Learning Engineer

Waymo