51 to 75 of 130 JAX Jobs in the UK

equivalent practical experience. Strong Python programming and software-engineering skills. Experience developing machine-learning, generative AI or LLM applications. Practical experience with PyTorch, JAX, TensorFlow or a similar framework. Understanding of agentic AI concepts such as tool calling, planning, memory and workflow orchestration. Knowledge of reinforcement learning, preference optimisation ...

agents, LLM research, as well as reinforcement learning and fine-tuning Strong knowledge of linear algebra, calculus and statistics.Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas.Depending on your experience: project supervision, leadership, or management.Strong knowledge of LLM landscape today ...

record of shipping products and features inside a fast-moving environment Strong coding skills in Python and familiarity with ML frameworks like PyTorch or JAX Experience training, fine-tuning, and evaluating large language models Deep familiarity with large language models and web agents, how they work, and how they ...

. Experience limited to API usage or prompt engineering is not sufficient. Experience building large-scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks. Deep knowledge of large-model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking. Evidence of research/ ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

especially knowledge in formal methods, stochastic systems and processes, the foundations of deep learning. Experience coding with deep learning libraries such as Pytorch/JAX is essential. Applicants must hold a PhD in computer science, mathematics or equivalent experience.Please see job description for a full list of requirements. What ...

. Experience limited to API usage or prompt engineering is not sufficient. Experience building large‐scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks. Deep knowledge of large‐model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking. Evidence of research/ ...

series). Experience limited to API usage or prompt engineering is not sufficient.Experience building large-scale data pipelines and distributed training systems using PyTorch, JAX, or equivalent frameworks.Deep knowledge of large-model training and evaluation: optimization, parallelism, mixed precision, checkpointing, experiment design, ablations, and benchmarking.Evidence of research/technical quality ...

practical experience. 5 years of experience developing, training, and deploying diffusion models and generative video architectures. Experience writing custom GPU/TPU kernels (e.g., JAX or PyTorch) for model serving. Preferred qualifications: PhD in Computer Science, Machine Learning, AI, or a related quantitative field or equivalent practical experience. Experience ...

learning techniques, including designing new architectures, hands-on experimentation, analysis, and visualisation.Strong knowledge of linear algebra, calculus and statistics.Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas.A passion for applying ML research to real world problems.Depending on your experience: project ...

including designing new architectures, hands-on experimentation, analysis, and visualisation. Strong knowledge of linear algebra, calculus and statistics. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas. A passion for applying ML research to real world problems. Depending on your ...

language models, retrieval systems and domain-specific tools. Qualifications include a degree in CS or related field, strong Python, and experience with PyTorch/JAX/TensorFlow; familiarity with agentic AI concepts and RL is essential. #J-18808-Ljbffr ...

bringExperience building ML or AI systems in productionHands-on experience with LLMs and agentic AIStrong Python, Rust, C++ or JavaExperience with PyTorch, JAX or TensorFlowExposure to LangGraph, Google ADK or similarStrong understanding of ML fundamentals and model trade-offsExperience with MLOps, multimodal AI or constrained hardware is a bonus rather ...

Computer Science, Machine Learning, Physics, or a related field with a strong record of publishing novel research. Possesses hands‐on expertise in PyTorch or JAX developing modern architectures such as generative models, Bayesian optimization, or foundation models. Demonstrates experience applying machine learning to scientific domains or simulations with a deep ...

training RL agents and iterating on reward design, with strong intuitions for what works and why Proficiency in Python and ML frameworks (PyTorch or JAX) with the ability to implement and run experiments independently Track record of research output: publications, open-source contributions, or impactful technical projects Willingness to work ...

systems (bonus for low-resource languages). Deep knowledge of at least one neural TTS architecture and related vocoders. Proficiency with PyTorch, TensorFlow, or JAX for building and training models. Experience with audio processing tools (e.g., librosa, Praat, torchaudio). Linguistic & Cultural Sensitivity Experience working with multilingual or low-resource ...

optimizing model serving infrastructure or inference backends. Preferred qualifications: Experience with developing serving infrastructure. Experience programming hardware accelerators (GPUs, TPUs) via ML frameworks (e.g., JAX, PyTorch) or low-level programming models (e.g., Pallas, CUDA, OpenCL). Experience profiling software to identify performance bottlenecks. Experience with distributed ML systems optimization ...

About You Strong Python engineering skills and experience working in large codebases (testing, code review, CI, operational ownership). Hands‐on experience with PyTorch, JAX, or similar. Strong systems and infrastructure fundamentals. Experience with LLM training or post‐training: fine‐tuning, RL, distillation, evaluation, and/or data pipelines. Excellent ...

Succeed Experience building, training, adapting, or evaluating machine learning models. Strong software engineering skills in Python and modern ML frameworks such as PyTorch, JAX, or TensorFlow. Experience designing experiments, evaluation metrics, or test sets for model performance. Ability to debug model behavior using data, traces, logs, and qualitative feedback. Experience ...

adapting, or evaluating large-scale foundation models, and building data pipelines for heterogeneous datasets. 3 years of experience with modern machine learning frameworks (e.g., JAX, PyTorch, TensorFlow) and distributed training on accelerators. 3 years of experience designing evaluations, metrics, and controlled ablations for research projects. 3 years of experience designing ...

solid production track record Proficient in Python and at least one systems language - Rust, C++, or Go Hands-on experience with PyTorch or JAX in an industry setting Experience with inference frameworks: vLLM, SGLang, TensorRT-LLM Solid distributed systems fundamentals and experience operating production ML infrastructure Working knowledge of modern ...

engineers through technical leadership, code reviews and best practices. Building systems that balance performance, latency, reliability and scalability. Technology Python PyTorch and/or JAX GPU-based model training and inference Modern machine learning infrastructure What We're Looking For Commercial experience building and deploying machine learning models into production. ...

creative teams to turn ML capabilities into compelling experiences What We’re Looking For: Strong knowledge of Python Experience writing models in PyTorch or JAX A good understanding of how large language models work Comfortable working along the whole ML pipeline: from data preparation to model design, training and evaluation ...

generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. Familiarity with modern software engineering practices. You document experiments clearly and communicate trade‐offs crisply. ...