1 to 25 of 55 Google JAX Jobs

Senior Machine Learning Researcher - MSR AI for Science

Hiring Organisation
Microsoft
Location
Cambridge, Cambridgeshire, UK
Employment Type
Full-time
learning (model design, large-scale training, evaluation and reproducibility), statistics and linear algebra. Proficiency in Python and modern ML/scientific frameworks (e.g., PyTorch, JAX, TensorFlow, NumPy, SciPy, Pandas). Peer-reviewed publications in leading venues (e.g., NeurIPS, ICML, ICLR or leading journals). Excellent technical communication for collaborating ...

2026 Graduate Software Engineer - PyTorch

Hiring Organisation
Graphcore
Location
Newport, UK
Employment Type
Full-time
Team The Frameworks team ensures Graphcore hardware works seamlessly with the tools that ML engineers and researchers love – Pytorch, Triton and in future, Jax and TensorFlow. We're a talented and diverse team of engineers and we foster a culture of collaboration, openness and learning. All our software teams ...

2026 Graduate Software Engineer - PyTorch

Hiring Organisation
Graphcore
Location
Bradley Stoke, Gloucestershire, UK
Employment Type
Full-time
Team The Frameworks team ensures Graphcore hardware works seamlessly with the tools that ML engineers and researchers love – Pytorch, Triton and in future, Jax and TensorFlow. We're a talented and diverse team of engineers and we foster a culture of collaboration, openness and learning. All our software teams ...

Research Scientist | Language AI

Hiring Organisation
DeepL
Location
London, UK
Employment Type
Full-time
science Programming experience in Python and at least one other programming language Experience with a modern machine learning framework such as PyTorch, TensorFlow or JAX High level of intrinsic motivation and strong interest in Deep Learning Hands-on experience in Deep Learning through self-directed projects that go well ...

Research Scientist | Language AI

Hiring Organisation
DeepL
Location
Slough, Berkshire, UK
Employment Type
Full-time
science Programming experience in Python and at least one other programming language Experience with a modern machine learning framework such as PyTorch, TensorFlow or JAX High level of intrinsic motivation and strong interest in Deep Learning Hands-on experience in Deep Learning through self-directed projects that go well ...

PhD Position in Deep Learning for Acoustic Sensor Fusion on Intelligent Vehicles

Hiring Organisation
Delft University of Technology
Location
Delft, Zuid-Holland, Netherlands
Employment Type
Permanent
Salary
EUR Annual
artificial intelligence, machine learning, intelligent vehicles/robotics, acoustics and signal processing, computer vision. Demonstratable experience in applying Deep Learning, using PyTorch, TensorFlow, JAX on real-world sensor data. Experience with robotic/vehicle perception tasks with computer vision, lidar or radar. Experience in acoustic perception is a plus. ...

Quantitative Developer (Python) - Hybrid London - Up To 250k

Hiring Organisation
Opus Recruitment Solutions
Location
London, United Kingdom
Employment Type
Permanent
Salary
GBP 150,000 - 250,000 Annual
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 ...

Quantitative Developer Python - Hybrid London - Up To 250k!

Hiring Organisation
Opus Recruitment Solutions Ltd
Location
Central London, London, England, United Kingdom
Employment Type
Full-Time
Salary
£150,000 - £250,000 per annum
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 ...

Senior Deep Learning Researcher - Model Efficiency - Qualcomm - Amsterdam

Hiring Organisation
Qualcomm
Location
Amsterdam, Noord-Holland, Netherlands
Employment Type
Permanent
Salary
EUR Annual
Company Qualcomm Technologies Netherlands B.V. Job Area Engineering Group, Engineering Group > Machine Learning Researcher General Summary At Qualcomm AI Research, we are advancing AI to make its core capabilities - perception, reasoning, and action - ubiquitous across ...

Technical Lead, Research Engineer, AI for Chip Design

Hiring Organisation
Google DeepMind
Location
London, England, United Kingdom
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 … 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 ...

Founding Machine Learning Engineer

Hiring Organisation
Bjak
Location
Dublin, Ireland
Employment Type
Permanent
Salary
EUR 80,000 - 120,000 Annual
Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

Founding Machine Learning Engineer

Hiring Organisation
Bjak
Location
Etzelwang, Bayern, Germany
Employment Type
Permanent
Salary
EUR 80,000 - 110,000 Annual
Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

Senior Research Scientist | Voice

Hiring Organisation
DeepL
Location
London, UK
Employment Type
Full-time
Meet DeepL DeepL is a global communications platform powered by Language AI. Since 2017, we've been on a mission to break down language barriers. Our human-sounding translations and intelligent writing suggestions are designed ...

Machine Learning Consultant

Hiring Organisation
Lynx Recruitment Ltd
Location
Melbourn, Royston, Hertfordshire, England, United Kingdom
Employment Type
Full-Time
Salary
Salary negotiable
*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 ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Midlands, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Bradford, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Sheffield, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
London, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Nottingham, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Aberdeen, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Birmingham, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Swindon, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Belfast, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Southampton, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...

Senior Math Libraries Engineer, CPU and GPU Optimization

Hiring Organisation
NVIDIA
Location
Shrewsbury, Shropshire, UK
Employment Type
Full-time
software accelerated computing ecosystem is constantly evolving, including shifts towards hybrid backends, deep integration with high-level languages and ecosystems (such as Python, Numpy, JAX, MLIR...), and optimization at runtime for maximum flexibility and performance. Our libraries follow CUDA Everywhere approach to let developers use highly-optimized mathematical operations … architectures. Experience with CCCL, OpenMP, OpenACC, multi-threading, MPI, PGAS. Experience with scientific and deep learning libraries and frameworks such as PyTorch, JAX, MKL, MAGMA, PETSc, Kokkos, etc. With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world's most ...