1 to 25 of 87 JAX Jobs in London

keywords. Useful experience Experience with some of the following would be valuable, but we do not expect you to know everything: PyTorch, TensorFlow or JAX Hugging Face LLM APIs and open-source models RAG and vector databases Apache Airflow AWS S3 ECS/EKS Redshift Docker Kubernetes GitHub Actions Pulumi ...

with strong software engineering practicesDemonstrated expertise in building production-quality code and data pipelines for ML systemsProficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applicationsUnderstanding of LLM training methodologies including instruction fine-tuning, preference ...

C++ or Java A strong background in the mathematical foundations of core ML algorithms, paired with hands‐on experience using deep learning frameworks (PyTorch, JAX, TensorFlow), agentic orchestration ecosystems (LangGraph, Google ADK), and Model Context Protocol tooling (RustMCP) Strong written and verbal communication skills to document your work effectively ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Knowledge of command‐line tools and shell scripting. Knowledge of software engineering best practices, including continuous integration (CI) and continuous deployment (CD), containerization ...

track record of building and maintaining production‐quality systems. Expert‐level Python experience with numerical, statistical, or ML libraries (e.g. NumPy, pandas, PyTorch, JAX, TensorFlow). Deep understanding of statistical modelling, optimisation, or machine‐learning techniques. Experience building and operating backend systems and data pipelines in a cloud environment (e.g. ...

track record of building and maintaining production‐quality systems. Expert‐level Python experience with numerical, statistical, or ML libraries (e.g. NumPy, pandas, PyTorch, JAX, TensorFlow). Deep understanding of statistical modelling, optimisation, or machine‐learning techniques. Experience building and operating backend systems and data pipelines in a cloud environment (e.g. ...

language understanding. The ability to write clean and maintainable code in Python, Rust, C++ or Java Experience with frameworks such as PyTorch, Tensorflow or JAX Strong written and verbal communication skills to document your work effectively and present it to relevant stakeholders and users A passion for identifying and implementing ...

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, ...)Excellent communication skills to report and present research findings and developments clearly, both orally and in writing.Curious and innovative disposition capable of devising ...

EMNLP, NAACL, ICLR) ) with specific focus on training data curation, synthetic data generation, etc.* 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 ...

generative models. Strong publication record or history of impactful project delivery in RL or related areas. Proficiency in Python and standard ML frameworks (e.g., JAX, TensorFlow). Experience with large‐scale distributed training and data processing. Proven ability to lead complex and ambiguous technical projects from conception to completion. ...

high level familiarity with the architecture and operation of large language models. Have personally implemented models in common ML frameworks such as PyTorch, Jax or TensorFlow. Possess a strong foundation in data structures, algorithms, and software engineering principles. Exhibit excellent problem‐solving and analytical skills, with a proactive approach ...

computer science, or a related field. Deep practical experience in Python and at least one modern machine learning framework such as PyTorch, TensorFlow, or JAX, experience working with large compute clusters and ML infrastructure is a plus. A track record of leading self-directed research projects that go well beyond ...

computer science, or a related field. Deep practical experience in Python and at least one modern machine learning framework such as PyTorch, TensorFlow, or JAX, experience working with large compute clusters and ML infrastructure is a plus. A track record of leading self-directed research projects that go well beyond ...

physics, computer science, or a related field.Deep practical experience in Python and at least one modern machine learning framework such as PyTorch, TensorFlow, or JAX, experience working with large compute clusters and ML infrastructure is a plus.A track record of leading self-directed research projects that go well beyond academic ...

good fit if you:Are proficient in Python and async/concurrent programming with frameworks like TrioHave experience with machine learning frameworks (PyTorch, TensorFlow, JAX)Have industry experience in machine learning researchCan balance research exploration with engineering implementationEnjoy pair programming (we love to pair!)Care about code quality, testing ...

Mathematics, etc.) or equivalent practical experience.ML Fundamentals: Deep understanding of machine learning principles and techniques.Framework Expertise: Strong proficiency in deep learning frameworks such as JAX or PyTorch.Modern Architectures: Hands-on experience building and working with modern model architectures (e.g., Transformers, GNNs, Diffusion Models).Full ML Lifecycle: Experience taking models from ...

Infrastructure as Code, and DevOps practices Production support, monitoring, observability, and incident management Experience with AI/ML frameworks such as PyTorch, TensorFlow, or JAX Exposure to LLMs, RAG, Agentic AI, model serving, or MLOps environments What's on Offer? Work on innovative AI and Agentic AI solutions Hybrid working ...

Agents, reasoning models, and RAG Have high-level familiarity with the architecture and operation of large language models Have personally implemented models in PyTorch, Jax, or TensorFlow Have a strong foundation in data structures, algorithms, and software engineering principles Exhibit excellent problem-solving and analytical skills with a proactive approach ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Hands-on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud Storage, Artifact Registry ...

transformers, SSMs) Solid development skills in Python and/or C++ Familiarity with ML libraries/frameworks such as PyTorch, TensorFlow, and/or JAX Intellectual curiosity, versatility, and originality combined with a pragmatic outlook Ability to thrive in a collaborative, team-oriented environment Ability to reason through quantitative problems ...

Java, and their associated security considerations.Coding experience with AI & ML, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn.Familiarity with current and emerging regulations and standards, such as the EU AI Act, the Digital Services Act, the Digital Markets Act, GDPR ...

Legal, Compliance, and external partners like Google DeepMind.Skills and qualifications EssentialAI/ML Engineering: Deep conceptual and practical understanding of deep learning frameworks (e.g., JAX, PyTorch, TensorFlow), large-scale cloud training or inference infrastructure, and LLM ecosystems.Adversarial ML Mindset: Strong familiarity with AI security threat vectors (prompt injection, model inversion ...

prototypes to reliable deployment; Have experience partnering closely with researchers throughout implementation; Be proficient in Python and deep learning frameworks such as PyTorch or JAX; Have strong software engineering fundamentals, including clean, testable, well‐documented and maintainable code, version control and code reviews; Have experience with distributed training frameworks such ...

large, possibly noisy, data sets. Strong analytical and quantitative skills, and a detail-oriented mindset. Strong proficiency in machine learning libraries such as Torch, JAX or TensorFlow. Competence in Python, cluster environment, and general software engineering principles (source control, testing, collaborative workflow). Excellent written and verbal communication skills, willing ...

applied AI research. Commercial or research experience implementing machine learning algorithms. Strong programming skills in Python; C++ experience is advantageous. Experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks. Why Apply? Join a world-class AI research team working on cutting-edge machine learning technologies. Play a key role ...