10 of 10 Google JAX Jobs in the UK

modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training ...

Strong software engineering skills: clean code, testing, debugging, version control, and system design Proficiency in Python with experience in ML frameworks (PyTorch, TensorFlow, or JAX) Hands-on experience with cloud platforms (AWS, GCP, or Azure) and containerisation (Docker, Kubernetes) Solid understanding of ML fundamentals (model training, evaluation, common architectures ...

above) in Computer Science, Engineering, Physics, Maths or similar Strong Python skills Experience with ML frameworks such as PyTorch, TensorFlow and/or JAX Exposure to applying ML in novel domains or resource-constrained settings Familiarity with a range of ML techniques (e.g. CV, generative models, signal/audio ...

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. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics ...

. Domain Depth: Hands-on expertise in VLMs , Diffusion/Generative Video , or LLM Pre-training . Frameworks: Mastery of Python and PyTorch or JAX . Systems: Experience with distributed training and complex numerical debugging. Randstad Technologies Ltd is a leading specialist recruitment business for the IT & Engineering industries. ...

production systems Hands-on post-training experience (RLHF, DPO, or similar) and deployed LLMs in production Strong Python and modern ML tooling (PyTorch/JAX, training infrastructure, evaluation pipelines) Track record shipping research from prototype to production impact Comfortable owning work end-to-end: architecture to data pipelines ...

Silicon Valley office | Reports to Founders Introduction: Join us at Fastino as we build the next generation of LLMs. Our team, boasting alumni from Google Research, Apple, Stanford, and Cambridge is on a mission to develop specialized, efficient AI. Fastino's GLiNER family of open source models has been … comparable generative AI architectures Optional - Comprehensive technical proficiency and practical experience with leading deep learning frameworks, including advanced competency in one of PyTorch, JAX, TensorFlow, or equivalent platforms for model development and optimization ...

engineering, fine-tuning, RAG (Retrieval Augmented Generation) Multimodal AI (text, image, audio, video) Agentic frameworks and orchestration AI/ML Engineering Python, PyTorch, TensorFlow, JAX NLP, deep learning, model optimisation Evaluation frameworks and guardrails Data & Platforms Vector databases (Pinecone, Weaviate, FAISS, Milvus) Knowledge graphs and semantic search Data engineering …/CD for ML, model monitoring, versioning Kubernetes, Docker, serverless architectures MLflow, Kubeflow, SageMaker or similar Cloud & Architecture Azure AI, AWS AI/Bedrock, Google Vertex AI API-led architectures and microservices Security, governance, and responsible AI Frameworks & Tooling LangChain, LlamaIndex, Semantic Kernel or equivalent Integration with enterprise systems ...

with teleops teams to transform human trajectories into robust robotic skills through behavior cloning. High-Performance Engineering: Designing and profiling research-grade PyTorch/JAX code to support large-scale, distributed RL infrastructure. Essential Skills Needed Deep Learning Mastery: 5+ years building and shipping models, with deep hands … Expert: A proven track record of solving complex, real-world problems using Deep Reinforcement Learning. Technical Rigour: Mastery of Python and PyTorch/JAX, including the ability to profile performance and debug complex numerical stability issues. Robotics Foundation: Practical experience with simulators (Isaac Sim/MuJoCo) and a deep ...

Deep Learning Engineer About the Company Join a fast-growing AI-driven technology company that’s modernising prediction and decision systems for complex, real-world industries. The organisation builds powerful AI platforms that help partners ...