13 of 13 JAX Jobs in the South East

Python, R, C, C++, or Java, and their associated security considerations. • Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. • Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. • Strong understanding of modern More ❯

Python, R, C, C++, or Java, and their associated security considerations. * Previous experience with ML, LLM, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn is desirable. * Experience in implementing secure coding practices, DevOps, CI/CD pipelines and familiarity with secure software development life cycle (SDLC) methodologies. * Strong understanding of modern More ❯

enterprise API About You Master’s or PhD in Computer Science (or equivalent proven track record) 4 + years working on large-scale ML codebases Hands-on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed/FSDP/SLURM/K8s) Experience in deep learning, NLP or LLMs; bonus for CUDA or data-pipeline chops Strong software-design More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

may be a good fit if you: Have 5+ years of industry-related experience Are proficient in Python and have experience with deep learning frameworks such as PyTorch or Jax Have a strong software engineering background and are interested in working closely with researchers and other engineers Enjoy pair programming (we love to pair!) Care about code quality, testing, and More ❯

that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C/ More ❯

that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C/ More ❯

that values innovation, openness, and personal growth Skills & Experience: PhD in Computer Science, Electrical Engineering, or equivalent industry research experience Solid background in machine learning frameworks (PyTorch, TensorFlow, or JAX) Experience in efficient ML techniques: quantization, pruning, distillation Familiarity with deployment on mobile and embedded devices (e.G. Smartphones, mobile CPU/GPU/NPU) Strong skills in Python, C/ More ❯

and requirements for an end-to-end stack; Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Experience with data sampling and data curation pipelines for autonomous driving datasets Experience in state-of-the-art architectures for end-to-end autonomous driving Experience More ❯

and requirements for an end-to-end stack; Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Experience with data sampling and data curation pipelines for autonomous driving datasets Experience in state-of-the-art architectures for end-to-end autonomous driving Experience More ❯

and requirements for an end-to-end stack; Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Experience with data sampling and data curation pipelines for autonomous driving datasets Experience in state-of-the-art architectures for end-to-end autonomous driving Experience More ❯

The world will be unrecognisable in 5 years. Machine learning models are driving our cars, testing our eyesight, detecting our cancer, giving sight to the blind, giving speech to the mute, and dictating what we consume, enjoy, and think. These More ❯

latency ML pipelines in high-performance environments is a strong plus Strong engineering skills, including Python, CUDA, or C++ Knowledge of machine learning frameworks such as PyTorch, TensorFlow, or JAX Proficiency in GPU programming for training and inference acceleration (e.g., CuDNN, TensorRT) Experience with distributed training for scaling ML workloads (e.g., Horovod, NCCL) Exposure to cloud platforms and orchestration tools More ❯