1 to 25 of 31 Remote/Hybrid Google JAX Jobs in the UK

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 ...

Senior) or 8+ years (Staff) of hands‐on applied ML experience, including production‐grade work. Deep familiarity with modern deep learning stack (PyTorch or JAX), and breadth across at least two of: generative models (diffusion, autoregressive, flow‐based), graph and equivariant networks, vision (CNNs, ViTs, multimodal LLMs), search ...

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 ...

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 ...

related frameworks• Experience on AI/ML model implementations using scikit learn, Tensor flow etc.,• Experience with Chat, IVR, Banking will be plus• Google Cloud (GCP) knowledgeAbout NTT DATANTT DATA is a $30 billion business and technology services leader, serving 75% of the Fortune Global 100. We are committed ...

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 ...

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 ...

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 ...

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 ...

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 ...

strong programming skills in Python , Rust, or similar; and strong software engineering fundamentals building performant and reliable systems. Proficient in deep learning frameworks (Pytorch, JAX, TensorFlow). You can work on different layers of the stack from low-level training backends, data ingestion to ML/RL algorithmic design ...

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 ...

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 ...

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 ...

ability to collaborate effectively in interdisciplinary teamsNice-to-haves:Experience in autonomous driving, robotics, or simulation systemsFamiliarity with large-scale training (e.g., FSDP, DeepSpeed, JAX)Experience with sim-to-real transfer or data-efficient learningContributions to open-source ML tools or research infrastructureWhat we offer you Attractive compensation with ...

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 ...

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 ...

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 ...

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 ...

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 … data generation, data curation, or curriculum learning pipelinesExperience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google CloudHands-on experience with MLOps, experiment tracking, and model deployment systemsStrong interest in staying current with ML research literature and ability to quickly implement ...

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 ...

have strong experience building production-grade machine learning systems at scale You are experienced with modern ML frameworks such as PyTorch, TensorFlow, or JAX You have worked with or are interested in multimodal machine learning You understand how to design systems that balance automation with quality, safety, and user ...

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 … writing production code.Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, MS Azure, LambdaLabs or Google Cloud.You will enjoy:Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods and technologies.Plenty ...

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 … writing production code. Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, Microsoft Azure, LambdaLabs or Google Cloud. You will enjoy Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods ...

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 … source libraries or writing production code. Experience training large‐scale models over distributed nodes with cloud tools such as Amazon AWS, MS Azure, or Google Cloud. You will enjoy Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting‐edge methods ...