76 to 100 of 130 JAX Jobs in the UK

application is highly valued.LLM experience is valuable, but not necessary.Strong engineering skills valued, especially any of: GPU kernel development, lower-level Pytorch/JAX development, intermediate C++, FPGA/ASIC experience, HPC experience.The estimated base salary range for this position is 200,000 to 300,000 USD per year ...

refine existing products.Prior experience in working cross-discipline in matrixed organizations. Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR.Experience using large-scale distributed training and inference.Strong mentorship, communication, and problem-solving skills.A demonstrated passion for applied ML models and products.Bonus: experience ...

refine existing products. Prior experience in working cross-discipline in matrixed organizations. Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR. Experience using large-scale distributed training and inference. Strong mentorship, communication, and problem-solving skills. A demonstrated passion for applied ...

Mathematics, or a related field or equivalent experience. Proficiency coding in Python or similar languages, as well as machine learning and analysis tooling (e.g., JAX, PyTorch, Pandas). Experience working with large language models, human-in-the-loop data annotation systems, or AI evaluation frameworks. A willingness to embrace ...

high agency and low egoStrong candidates may also haveExperience with large-scale distributed training (RL, pre-training, or post-training)Familiarity with JAX, PyTorch, or similar ML frameworksA track record of operating at the edge of research and infra in a fast-moving environmentDeadline to apply: None. Applications will ...

compiler, workflow engine, distributed training system, database, debugger, or orchestration platform Meaningful contributions to systems such as Ray, Temporal, Kafka, Erlang/BEAM, PyTorch, JAX, TensorFlow, or comparable projects Strong API design judgement, open-source collaboration, and disciplined approaches to testing, versioning, compatibility, and reproducibility Experience in ML systems ...

running experiments, and debugging pipelines, and who can carry research results through to production with engineering. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow). Ability to communicate clearly, collaborate across teams, and align research work with product and engineering priorities. Nice to have Experience quantifying uncertainty ...

experience with other robotics simulators (Drake, MuJoCo, Newton, Isaac, RaiSim, Jiminy, Brax). PyTorch, our user-facing interface. Other JIT compilers such as Triton, JAX, or Numba. 3D tooling and interactive editors (Maya, Blender, Houdini). Serialization formats (USD, glTF, MJCF/URDF). C++, CUDA, or Metal. Developer relations ...

research direction with strong execution, and experience mentoring others on a fast‐moving, applied research team. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Nice to have Demonstrated experience fine‐tuning and training ...

silicon. What they’re looking for: Degree in Physics/Engineering/Electronics and a PhD in Machine Learning Strong Python + PyTorch/JAX skills Publication record in ML or EDA venues Ability to move between research and implementation Clear communication across ML and circuit‐design audiences Experience modelling ...

skills. Ability to balance strategic thinking with hands-on technical leadership. Comfortable operating within a fast-paced, high-growth environment. Technology Environment Python PyTorch JAX Large Language Models (LLMs) GPU Training & Inference Infrastructure Distributed ML Systems Why Join? Help shape the future of AI-powered productivity. Work alongside a small ...

models — architecture, training, and inference. Experience solving real problems using reinforcement learning with deep neural networks in any domain. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You are self‐driven, pro‐active, communicate efficiently, document experiments clearly and communicate trade‐offs ...

lead complex research efforts while staying grounded in product impact, user experience, and real-world performance. Strong coding and experimentation skills (Python, PyTorch/JAX, audio processing libraries). Ability to communicate clearly, collaborate across teams, and align research work with product and engineering priorities. Proven experience mentoring others ...

integrating ML systems into production while staying grounded in product impact and real-world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

silicon. What they’re looking for: First degree in relevant STEM subject 3+ years in industry building production ML systems Python + PyTorch/JAX expertise for training, inference, and model export Experience building data pipelines for complex scientific/engineering datasets Solid software engineering fundamentals - testing, CI/ ...

evaluation, and testing Optimise reliability, scalability, latency, and user experience Work closely with Product and Engineering teams to solve complex customer problems Python PyTorch JAX vLLM Vector Databases What We're Looking For Strong software engineering fundamentals Experience building AI agents, copilots, RAG systems, or workflow automation platforms Excellent Python ...

Languages* Machine Learning Models* Model Training* Engineering Workflows* Data Analysis at Scale* Data Visualization* Data Ingestion Pipelines* Large Language Models (LLMs)* AI Agents* Tools* JAX* PyTorch #J-18808-Ljbffr ...

integrating ML systems into production while staying grounded in product impact and real‐world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

models, fighting deep learning compilers). Excellent understanding of GPU architecture and why some models run faster than others. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice to have: Robotics or autonomous ...

impacting real world drug design programs.Skills and qualifications Essential:Significant experience with large scale distributed training of LLMs.Experience with deep learning ML frameworks (either JAX or PyTorch).Knowledge of parallelism strategies and collective communication libraries (e.g. NCCL).Good understanding of GPU architectures. Reasoning about performance concepts is more important than ...

Senior Engineering Manager - Machine Learning United Kingdom We are on a mission to pioneer the world’s next era of play. As we grow across Europe and Latin America, we’re building The Playstack - the ...

Senior Engineering Manager - Machine Learning United Kingdom We are on a mission to pioneer the world’s next era of play. As we grow across Europe and Latin America, we’re building The Playstack - the ...

data quality, and ensure continuous training cycles. Preferred qualifications include 3+ years in deep learning, proficiency in Python and frameworks like PyTorch/JAX, and effective communication. The role offers 23 days off, private healthcare, and a pension scheme. #J-18808-Ljbffr ...

from data to UI. Applicants should have strong quantitative skills and experience in Python/SQL/TypeScript and PyData stack, plus PyTorch/JAX/TF for model development. #J-18808-Ljbffr ...

live training/matchday schedule, not just be technically neat. Python/SQL/TypeScript, PyData stack (Pandas/Polars, Altair, etc.), PyTorch/JAX/TF. You'll own things end-to-end, data through UI. Looking for strong quant background, first-principles engineering judgment, and taste ...