6 of 6 JAX Jobs in Slough

in AI research and applied machine learning. Expertise in key data science technologies and techniques, such as Python, Git, AWS, AWS Sagemaker, PyTorch, TensorFlow, Jax, Numpy, Scikit-learn, Time-series forecasting, Classification, Regression, Large-language models, and Experiment Design. Strong stakeholder management skills and the ability to communicate complex ideas More ❯

e.g., PyTorch, TensorFlow) • Solid grasp of ML techniques; independent and team-oriented mindset Preferred: • Experience in LLMs, reinforcement learning, or chip design • Familiarity with JAX and optimization frameworks If interested please apply below or send your CV to Trei@microtech-global.com More ❯

Learning, Computer Science, or related field Proven track record with LLMs, generative AI, optimization, or sequential learning Experience with deep learning frameworks (TensorFlow, PyTorch, JAX, etc.) Expertise in one or more: reinforcement learning, Bayesian optimization, optimal experimental design, agents, or causality Experience in drug/target discovery, treatment modeling, or More ❯

field • 5+ years of experience deploying AI/ML models in production settings • Proficient in Python, C++, and deep learning frameworks like PyTorch or Jax • Skilled in distributed training (e.g., DDP, FSDP) and model performance optimization • Experience with GPU architectures, cloud platforms, and hardware tradeoffs • Familiarity with low-level hardware More ❯

focus on GenAI 🚀 Strong expertise in Stable Diffusion, Flux, and similar architectures 🚀 Experience fine-tuning and optimising visual models 🚀 Hands-on experience with PyTorch, JAX, or TensorFlow 🚀 Proficiency in GPU inference optimisation (CUDA, BitsAndBytes, PEFT) 🚀 Experience deploying ML models using Kubernetes and/or Docker 🚀 Strong experience with Python and More ❯

collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology in More ❯