14 of 14 Remote JAX Jobs in London

Science, Machine Learning, Mathematics, or a closely related field. Strong foundational knowledge of machine learning and deep learning , with hands-on experience using frameworks such as PyTorch , TensorFlow , or JAX . Exposure to LLM fine-tuning and/or experience working with transformer-based models (e.g. GPT, LLaMA, Mistral). Interest in or familiarity with agentic AI frameworks and multi More ❯

or engineering roles. Strong expertise in Machine Learning and Deep Learning, with exposure to Reinforcement Learning as a plus. Proficiency in Python and modern ML libraries (e.g., TensorFlow, PyTorch, JAX, or Keras). Experience with version control systems (GitHub, GitLab) and knowledge of clean, maintainable coding practices. Familiarity with CI/CD pipelines for automating ML workflows. Ability to thrive More ❯

our enterprise API About You Masters 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 ❯

language processing and computer vision) in industry. Master, PhD degree or equivalent experience in applied mathematics, computer science, bioinformatics or related scientific fields. Proficiency in software engineering (Python, Pytorch, JAX, Docker, Linux). A willingness to learn and develop skills in computational genomics including building an understanding of biological concepts related to cancer. Excellent communication skills in English. Appropriate work More ❯

NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, ICLR with specific focus on agent systems, tool use, or multi-agent coordination. Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ). Excellent communication skills to report and present research findings and developments clearly, both More ❯

e.g., NeurIPS, ICML, ICLR, ACL,EMNLP, NAACL, ICLR) with specific focus on agent systems, tool use, or multi-agentcoordination Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Excellent communication skills to report and present research findings anddevelopments clearly, both orally and in writing. Curious and innovative disposition capable of devising novel, well-founded algorithmicsolutions to relevant More ❯

learn, etc.). Comfort working with large and complex datasets, including data wrangling and exploratory analysis. Experience (or strong interest) in GPU-powered model development (e.g. using PyTorch, TensorFlow, JAX, or similar). Demonstrated ability to work independently on real-world projects. Strong communicator, with the ability to share ideas, take feedback, and collaborate effectively. A master’s or PhD More ❯

learn, etc.). Comfort working with large and complex datasets, including data wrangling and exploratory analysis. Experience (or strong interest) in GPU-powered model development (e.g. using PyTorch, TensorFlow, JAX, or similar). Demonstrated ability to work independently on real-world projects. Strong communicator, with the ability to share ideas, take feedback, and collaborate effectively. A master’s or PhD 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 ❯

tools, and hardware optimisation techniques Experience with distributed training frameworks (Ray, Dask, PyTorch Lightning, etc.) Expertise with Python and/or C/C++ Development with machine learning frameworks (JAX, Tensorflow, PyTorch etc.) Passion for profiling, identifying bottlenecks, and delivering efficient solutions. Highly Desirable Track record of successfully scaling ML models. Experience writing custom CUDA kernels or XLA operations. Understanding More ❯

Research Engineer, Machine Learning (Horizons) London, UK About Anthropic Anthropics mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team is 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 . 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 . More ❯

Life at UiPath The people at UiPath believe in the transformative power of automation to change how the world works. We're committed to creating category-leading enterprise software that unleashes that power. To make that happen, we need people More ❯