5 of 5 Permanent JAX Jobs in Slough

pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling ...

pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling ...

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

transformers, diffusion, retrieval-augmented systems, reinforcement learning, etc.). Strong coding skills in Python and experience with at least one major ML framework (PyTorch, JAX, TensorFlow). Ability to bridge high-level research with practical implementation. Curious, humble, and excited to work on hard technical problems with real enterprise impact. ...

software and research teams. Requirements Strong background in machine learning algorithm development and optimisation. Proficiency in Python and modern ML frameworks such as PyTorch, JAX or TensorFlow. Experience with data modelling, preprocessing and performance benchmarking. Understanding of ML evaluation metrics and model analysis methods. Excellent communication skills and collaborative mindset. ...