Jax Jobs
139 open roles, updated daily
Filters2
Apply for the Applied AI Engineer role at bjakcareer in China. Build proactive AI applications and reliable workflows for global users.
Apply for the Applied AI Engineer role at bjakcareer in China. Build proactive AI applications and reliable workflows for global users.
Join Ai2 in Seattle as a Research Engineer for a 3-year climate modeling project. Work with JAX, Google Cloud, and ML models. $146k-$220k.
Join Ai2 in Seattle as a Research Engineer for a 3-year climate modeling project. Work with JAX, Google Cloud, and ML models. $146k-$220k.
Join Waymo in Mountain View as a Perception Machine Learning Intern. Design multi-modality models using large-scale driving data.
Join Waymo in Mountain View as a Perception Machine Learning Intern. Design multi-modality models using large-scale driving data.
Apply for the Researcher, Efficient Inference role at makermaker.ai in San Francisco. Focus on quantization, speculative decoding, and production ML.
Apply for the Researcher, Efficient Inference role at makermaker.ai in San Francisco. Focus on quantization, speculative decoding, and production ML.
Join Ai2 in Seattle as a Research Scientist for a 3-year project developing an AI-powered climate model. Base salary $167,030 - $260,570.
Join Ai2 in Seattle as a Research Scientist for a 3-year project developing an AI-powered climate model. Base salary $167,030 - $260,570.
Join Waymo as a Summer Intern in Software Engineering for Sys Intel & Machine Learning in Mountain View, CA. Help optimize autonomous driving models.
Join Waymo as a Summer Intern in Software Engineering for Sys Intel & Machine Learning in Mountain View, CA. Help optimize autonomous driving models.
Lead generative chemistry AI models at Flagship Pioneering. Drive multi-modal drug discovery, manage computational teams, and shape venture strategy.
Lead generative chemistry AI models at Flagship Pioneering. Drive multi-modal drug discovery, manage computational teams, and shape venture strategy.
Join Mistral AI in Singapore as an Applied Science Research Engineer in Robotics to build and deploy advanced VLM and VLA models.
Join Mistral AI in Singapore as an Applied Science Research Engineer in Robotics to build and deploy advanced VLM and VLA models.
Apply for the Research Engineer - Auto Research role at Chai Discovery in San Francisco, building agent-powered systems for AI drug discovery.
Apply for the Research Engineer - Auto Research role at Chai Discovery in San Francisco, building agent-powered systems for AI drug discovery.
Apply for the Research Engineer - ML Infrastructure role at Chai Discovery in San Francisco. Build core machine learning training stacks and systems.
Apply for the Research Engineer - ML Infrastructure role at Chai Discovery in San Francisco. Build core machine learning training stacks and systems.
Join Waymo as a PhD Software Engineering Intern in Simulation. Work on uncertainty-aware algorithms in a hybrid role based in Mountain View.
Join Waymo as a PhD Software Engineering Intern in Simulation. Work on uncertainty-aware algorithms in a hybrid role based in Mountain View.
Lead AI and data science strategy at Xero in Melbourne. Oversee machine learning, analytics, and agentic workflows in a hybrid executive role.
Lead AI and data science strategy at Xero in Melbourne. Oversee machine learning, analytics, and agentic workflows in a hybrid executive role.
Build scalable AI applications and intelligent agents at Writer. Five years of production ML experience required. Hybrid role in London.
Build scalable AI applications and intelligent agents at Writer. Five years of production ML experience required. Hybrid role in London.
Join Waymo as a Machine Learning Engineering Intern in San Francisco. Train multi-task transformers on TPUs for autonomous driving.
Join Waymo as a Machine Learning Engineering Intern in San Francisco. Train multi-task transformers on TPUs for autonomous driving.
Join Writer as an AI Research Scientist in San Francisco or New York. Drive LLM research, agentic workflows, and production deployments.
Join Writer as an AI Research Scientist in San Francisco or New York. Drive LLM research, agentic workflows, and production deployments.
Join Writer as an AI Engineer in San Francisco, New York, or Seattle. Build scalable enterprise LLMs and AI agents with a $198K-$259K salary range.
Join Writer as an AI Engineer in San Francisco, New York, or Seattle. Build scalable enterprise LLMs and AI agents with a $198K-$259K salary range.
Join Waymo as a Research Scientist in World Model Post-Training. Conduct applied research in Mountain View, San Francisco, or New York.
Join Waymo as a Research Scientist in World Model Post-Training. Conduct applied research in Mountain View, San Francisco, or New York.
Apply for the Senior Manager, Applied AI Research role at MoneyLion in Prague or remote. Lead AI initiatives protecting 500 million users worldwide.
Apply for the Senior Manager, Applied AI Research role at MoneyLion in Prague or remote. Lead AI initiatives protecting 500 million users worldwide.
Join Discord as a Staff ML Engineer on the Consumer Revenue team. Build large-scale recommendation systems with a base salary of $272k-$340k.
Join Discord as a Staff ML Engineer on the Consumer Revenue team. Build large-scale recommendation systems with a base salary of $272k-$340k.