7 of 7 Remote/Hybrid Google JAX Jobs in the UK excluding London

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

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

related field. Research experience in at least one of: probabilistic machine learning, diffusion/flow‐based models. Proficiency in modern ML toolchains (e.g., PyTorch, JAX) and reproducible research practices. A track record of high‐quality publications, e.g. at ICML, NeurIPS, ICLR, AISTATS, ACL, EMNLP, CVPR, JMLR, Machine Learning ...

About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat’spossible in asset management. Today, AI is reshaping every aspect of our industry, from ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

ensuring models stay honest against field conditions and remain deployable in real-time systems. Experience with time-series or biosignals ML using PyTorch or JAX is highly valued, with hybrid work across Manchester and Los Angeles. #J-18808-Ljbffr ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefine what's possible ...