JAX Jobs
191 open roles, updated daily
Filters1
Apply for the 2027 Summer Intern, PhD, Research, AV Planning role at Waymo in Mountain View, CA. Hybrid onsite internship for doctoral students.
Apply for the 2027 Summer Intern, PhD, Research, AV Planning role at Waymo in Mountain View, CA. Hybrid onsite internship for doctoral students.
Waymo seeks a PhD intern in Mountain View or San Francisco for research in post-training autonomous driving AI. $85/hour hourly pay.
Waymo seeks a PhD intern in Mountain View or San Francisco for research in post-training autonomous driving AI. $85/hour hourly pay.
Apply for the Machine Learning Engineer role at Advanced Space in Westminster, CO. Hybrid position, salary $100K-$134K. Space exploration missions.
Apply for the Machine Learning Engineer role at Advanced Space in Westminster, CO. Hybrid position, salary $100K-$134K. Space exploration missions.
Apply for the Senior Machine Learning Engineer role at Graphcore in Cambridge, Bristol, or London. Scale AI models on specialised hardware.
Apply for the Senior Machine Learning Engineer role at Graphcore in Cambridge, Bristol, or London. Scale AI models on specialised hardware.
Apply for the Senior Machine Learning Engineer role at Graphcore in London, Bristol, or Cambridge. Scale AI models on specialized hardware.
Apply for the Senior Machine Learning Engineer role at Graphcore in London, Bristol, or Cambridge. Scale AI models on specialized hardware.
Apply for the Senior Machine Learning Engineer role at Graphcore in Bristol, London or Cambridge. Scale AI models on specialized hardware.
Apply for the Senior Machine Learning Engineer role at Graphcore in Bristol, London or Cambridge. Scale AI models on specialized hardware.
Apply for the Research Engineer, Safety & Alignment role at Cognition in San Francisco. Build safety evals and alignment methods for AI agents.
Apply for the Research Engineer, Safety & Alignment role at Cognition in San Francisco. Build safety evals and alignment methods for AI agents.
Apply for the Perception Intern role at Mach9 in San Francisco. Work on 2D image and 3D point cloud models for geospatial data analysis.
Apply for the Perception Intern role at Mach9 in San Francisco. Work on 2D image and 3D point cloud models for geospatial data analysis.
Apply for the 2026 Summer Intern, PhD, Research, World Modeling Evaluation role at Waymo in Mountain View, CA. Hybrid onsite PhD internship.
Apply for the 2026 Summer Intern, PhD, Research, World Modeling Evaluation role at Waymo in Mountain View, CA. Hybrid onsite PhD internship.
Apply for the 2027 Summer ML Internship at Waymo in Mountain View, CA. Hybrid PhD role focusing on planning and prediction at $85/hour.
Apply for the 2027 Summer ML Internship at Waymo in Mountain View, CA. Hybrid PhD role focusing on planning and prediction at $85/hour.
Join Airwallex in Singapore as a Senior Software Engineer for ML Platform. Build risk decisioning infrastructure. On-site role (5 days/week).
Join Airwallex in Singapore as a Senior Software Engineer for ML Platform. Build risk decisioning infrastructure. On-site role (5 days/week).
Apply for the Research Engineer role at Clera in Zürich. Build ML pipelines and post-training infrastructure. Competitive pay and visa sponsorship.
Apply for the Research Engineer role at Clera in Zürich. Build ML pipelines and post-training infrastructure. Competitive pay and visa sponsorship.
Apply for the GPU Kernel Engineer role at Sciforium in San Francisco, CA. Salary: $190K–$250K with equity. Work on custom GPU kernels for AI systems.
Apply for the GPU Kernel Engineer role at Sciforium in San Francisco, CA. Salary: $190K–$250K with equity. Work on custom GPU kernels for AI systems.
Apply for the Applied Machine Learning Engineer role at tracelabs in the United States. $185K–$245K salary plus equity.
Apply for the Applied Machine Learning Engineer role at tracelabs in the United States. $185K–$245K salary plus equity.
Apply for the RL Systems Infra role at Thinking Machines in San Francisco. Distributed training, RL workloads, $425K-$475K salary, visa support.
Apply for the RL Systems Infra role at Thinking Machines in San Francisco. Distributed training, RL workloads, $425K-$475K salary, visa support.
Apply for the AI Research Member of Technical Staff role at psi in Boston. Build AI agents and training systems to discover physics at scale.
Apply for the AI Research Member of Technical Staff role at psi in Boston. Build AI agents and training systems to discover physics at scale.
Apply for Member of Technical Staff, Applied AI at psi in Boston. Build AI systems to discover new physics at scale. Check requirements and apply.
Apply for Member of Technical Staff, Applied AI at psi in Boston. Build AI systems to discover new physics at scale. Check requirements and apply.
Apply for the Senior Software Engineer role at Planet in Ljubljana. Hybrid position focusing on IAM services, Java, and cloud platforms.
Apply for the Senior Software Engineer role at Planet in Ljubljana. Hybrid position focusing on IAM services, Java, and cloud platforms.
Apply for the 2027 Summer Intern, Perception Foundation Models role at Waymo in Mountain View, CA. Hybrid onsite graduate machine learning internship.
Apply for the 2027 Summer Intern, Perception Foundation Models role at Waymo in Mountain View, CA. Hybrid onsite graduate machine learning internship.
Build ML infrastructure for AI systems at Trajectory in San Francisco. Focus on training, inference, and GPU kernels. Apply on the official careers page.
Build ML infrastructure for AI systems at Trajectory in San Francisco. Focus on training, inference, and GPU kernels. Apply on the official careers page.
Apply for the PhD Machine Learning Software Engineer Intern role at Lyft in San Francisco, CA. Summer 2027 hybrid internship focusing on applied AI research.
Apply for the PhD Machine Learning Software Engineer Intern role at Lyft in San Francisco, CA. Summer 2027 hybrid internship focusing on applied AI research.
Apply for the LLM Scientist, AI for Science role at Mistral in Paris. Develop LLMs and deep learning frameworks for biology, chemistry, and physics.
Apply for the LLM Scientist, AI for Science role at Mistral in Paris. Develop LLMs and deep learning frameworks for biology, chemistry, and physics.
Apply as a Senior Machine Learning Scientist at Flagship Pioneering in Cambridge, UK. Build models and AI agents for somatic genomics.
Apply as a Senior Machine Learning Scientist at Flagship Pioneering in Cambridge, UK. Build models and AI agents for somatic genomics.
Apply for the Senior HPC & GPU Infrastructure Engineer role at Sciforium in San Francisco, CA. Manage high-density GPU clusters with a $150K-$220K salary.
Apply for the Senior HPC & GPU Infrastructure Engineer role at Sciforium in San Francisco, CA. Manage high-density GPU clusters with a $150K-$220K salary.