1 to 25 of 38 JAX Jobs in the East of England

Optimization context (Gaussian processes, Bayesian Neural Networks) on small and large datasets. Strong experience in at least one ML framework (PyTorch/TensorFlow/Jax) and robust experience in Python data science ecosystem (Numpy, SciPy, Pandas, etc.) Experience using a cloud computing service to reduce runtime to train and evaluate More ❯

a quantitative field (e.g., Computer Science, Machine Learning, Statistics, Physics, Mathematics) or equivalent industry experience. Strong background in deep learning frameworks (e.g., PyTorch, TensorFlow, JAX). Experience with distributed computing platforms (AWS, GCP, Azure, or on-prem clusters). Ideal: Kubernetes and Docker experience for scalable, reproducible workflows. Working at More ❯

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU, NPU 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 #J-18808-Ljbffr 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 #J-18808-Ljbffr 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 #J-18808-Ljbffr 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 #J-18808-Ljbffr 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 #J-18808-Ljbffr 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 #J-18808-Ljbffr 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 #J-18808-Ljbffr 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 #J-18808-Ljbffr More ❯

frameworks (e.g., PyTorch, TensorFlow)Solid grasp of ML techniques; independent and team-oriented mindset Preferred : Experience in LLMs, reinforcement learning, or chip designFamiliarity with JAX and optimization frameworks Why Join Us: Work on impactful AI research with real-world applicationsCollaborate with top researchers in a global teamFlexible hybrid work environmentOpportunities 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 Why Join Us: Work on impactful AI research with real-world applications Collaborate with top researchers in a global team Flexible hybrid 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 #J-18808-Ljbffr More ❯

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU, NPU More ❯

of the latest trends in AI/ML research and development Good familiarity with one of the popular ML frameworks such as PyTorch, TensorFlow, JAX, ONNX, or PaddlePaddle A history of developer engagement or participation in an active open-source development community Experience in Product Owner roles, or a desire More ❯

e.g., generative models, GNNs, active learning, foundation models) in a scientific context. Strong coding skills and experience with relevant ML frameworks (e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit). Deep understanding of molecular representation techniques and their application in modeling. Experience in designing and implementing robust, scalable AI More ❯

e.g., generative models, GNNs, active learning, foundation models)in a scientific context.- Strong coding skills and experience with relevant ML frameworks(e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit).- Deep understanding of molecular representation techniques and their application in modeling.- Experience in designing and implementing robust, scalable AI More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, basildon col-narrow-left Client: Founding Teams Location: basildon, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, hemel hempstead col-narrow-left Client: Founding Teams Location: hemel hempstead, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, bedford col-narrow-left Client: Founding Teams Location: bedford, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, ipswich col-narrow-left Client: Founding Teams Location: Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 Posted: 16.06.2025 Expiry More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, stevenage col-narrow-left Client: Founding Teams Location: Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 Posted: 16.06.2025 Expiry More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, watford, hertfordshire col-narrow-left Client: Founding Teams Location: watford, hertfordshire, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job More ❯

Social network you want to login/join with: Junior Machine Learning Engineer - AI startup, colchester col-narrow-left Client: Founding Teams Location: colchester, United Kingdom Job Category: Other - EU work permit required: Yes col-narrow-right Job Views: 4 More ❯