JAX Jobs
53 open roles, updated daily
Filters3
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.
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 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 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.
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 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.
Apply for the Senior Software Engineer, Vehicle Interaction role at Waymo in Mountain View, CA. Base salary $213,000 — $263,000 USD.
Apply for the Senior Software Engineer, Vehicle Interaction role at Waymo in Mountain View, CA. Base salary $213,000 — $263,000 USD.
Apply for the Senior Computer Vision Engineer role at Anduril Industries in Irvine, CA. Own the ML stack, edge compute models, and defense technology.
Apply for the Senior Computer Vision Engineer role at Anduril Industries in Irvine, CA. Own the ML stack, edge compute models, and defense technology.
Apply for the Research Engineer, Agentic EDA role at NormalComputing in New York City. Build multi-agent LLM systems. Salary: $200K-$400K + Equity.
Apply for the Research Engineer, Agentic EDA role at NormalComputing in New York City. Build multi-agent LLM systems. Salary: $200K-$400K + Equity.
Apply for the CAD Automation Researcher role at universalagi in San Francisco. Build AI foundation models for CAD geometry and physics simulation.
Apply for the CAD Automation Researcher role at universalagi in San Francisco. Build AI foundation models for CAD geometry and physics simulation.
Join ACTAI as a Senior Machine Learning Engineer in Spain. Build core ML systems and own production-ready AI applications end-to-end.
Join ACTAI as a Senior Machine Learning Engineer in Spain. Build core ML systems and own production-ready AI applications end-to-end.
Apply for the Senior Machine Learning Engineer role at ACTAI in Poland. Build core ML systems for proactive AI products.
Apply for the Senior Machine Learning Engineer role at ACTAI in Poland. Build core ML systems for proactive AI products.
Join bjakcareer in Zurich as a Senior Machine Learning Engineer. Own production ML systems and build proactive AI applications.
Join bjakcareer in Zurich as a Senior Machine Learning Engineer. Own production ML systems and build proactive AI applications.
Join ACTAI in the United States as a Senior Machine Learning Engineer to build proactive AI systems. Apply on the official careers page.
Join ACTAI in the United States as a Senior Machine Learning Engineer to build proactive AI systems. Apply on the official careers page.
Apply for Senior Machine Learning Engineer at bjakcareer in China. Build proactive AI apps and own critical ML subsystems in production.
Apply for Senior Machine Learning Engineer at bjakcareer in China. Build proactive AI apps and own critical ML subsystems in production.
Apply for the Senior Machine Learning Engineer role at bjakcareer in the United Kingdom. Build core ML systems for proactive AI products.
Apply for the Senior Machine Learning Engineer role at bjakcareer in the United Kingdom. Build core ML systems for proactive AI products.