14 of 14 Google JAX Jobs

processing libraries (e.g., Pandas, Polars, NumPy). Hands-on experience in building and maintaining automated data pipelines for large-scale data processing. Familiarity with machine learning frameworks (e.g., PyTorch, JAX, scikit-learn) as applied to data quality and augmentation tasks. Expertise in working with healthcare data, including familiarity with the OMOP Common Data Model (OMOP CDM). Strong experience More ❯

Computer Science, Machine Learning, Mathematics or similar field. * Experience building, deploying and maintaining NLP/Deep Learning Systems. * Extensive experience in at least one Deep Learning framework (PyTorch, TensorFlow, JAX). * Well versed in the scientific Python ecosystem (NumPy, Scikit-Learn, Pandas etc.). * Strong Data Engineering underpinnings and an ability to work with big data (Tbs). * Strong More ❯

or related field Strong proficiency in Python Proficiency in C++ Experience with Linux operating systems Must be a fast learner Desirable Exposure to machine learning frameworks such as PyTorch, JAX, Triton, TensorFlow Experience with distributed workload management systems such as Kubernetes, VLLM, Keras or MLOps pipelines Experience with hardware architectures or compilers Experience with test automation in a CI More ❯

validated. Technical Experience: Strong software engineering skills with experience in machine learning, numerical computing, and/or applied ML research Proficiency with Python and numerical + ML frameworks (PyTorch, JAX, numpy), familiarity with PyTorch 2.x compiler stack (e.g., TorchScript, TorchInductor) is a plus Practical experience deploying ML models or infrastructure in a research and production environment BS/MS More ❯

closely with researchers and product engineers to ship fast in an environment that iterates constantly What they’re looking for Strong Python engineering background Experience with frameworks like PyTorch, JAX or TensorFlow Demonstrated ability to scale ML workflows in real-world settings (cloud, GPU clusters, distributed training) Comfortable with MLOps tooling (W&B, Ray, Docker, etc.) Familiar with modern More ❯

also are on the look out for candidates who: Have deep familiarity with Python data ecosystem Understanding of Jupyter notebooks Exposure to machine learning libraries like PyTorch, XGBoost and JAX Understanding of crypto or traditional financial markets Strong API design and documentation skills What do you get in return? Up to 250k base (depending on experience) 3 days in More ❯

also are on the look out for candidates who: Have deep familiarity with Python data ecosystem Understanding of Jupyter notebooks Exposure to machine learning libraries like PyTorch, XGBoost and JAX Understanding of crypto or traditional financial markets Strong API design and documentation skills What do you get in return? Up to £250k base (depending on experience) 3 days in More ❯

This temporary role requires an ML Quant Engineer with expertise within an Investment Bank. The position is based in London and involves developing and implementing machine learning models to support financial decision-making. Client Details The hiring organisation is a More ❯

About Graphcore Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation of AI breakthroughs and power the widespread adoption of AI solutions More ❯

Snapshot Artificial Intelligence could be one of humanity’s most useful inventions. At Google DeepMind, we’re a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial intelligence. We use our technologies for widespread public benefit and scientific discovery, and collaborate with others on critical challenges, ensuring safety … been adopted by the HW team in multiple workstreams accelerating the Chip Design process. The mission of our team is to “Accelerate and Optimize Chip Design with AI.” At Google DeepMind we've built a unique culture and work environment where long-term ambitious research grounded in real problems can flourish. The Role As part of our team at … Google DeepMind you'll have opportunities to advance AI for Chip Design to enable breakthrough capabilities, and pioneer next-generation products in collaboration with major Product Areas. There are many fundamental research and transformative product landing opportunities, including but not limited to: Bring the most advanced ML models and technologies to Chip Design. Develop ML breakthroughs that will have More ❯

*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning Consultant, you will play a More ❯

AMLZ Recruiting is the executive search brand of AMLZ GmbH, headquartered in Wiesbaden, Germany. We specialize in placing exceptional professionals and executives with leading companies worldwide. Backed by a global network and deep industry expertise, we connect top talent with More ❯

AMLZ Recruiting is the executive search brand of AMLZ GmbH, headquartered in Wiesbaden, Germany. We specialize in placing exceptional professionals and executives with leading companies worldwide. Backed by a global network and deep industry expertise, we connect top talent with More ❯

Machine Learning Consultant As a Machine Learning Research Consultant you will tackle diverse, real-world challenges which are typically at the forefront of emerging AI/ML technologies. You will use your experience to design and deploy systems that apply More ❯