226 to 246 of 246 JAX Jobs in England

and requirements for an end-to-end stack. Experience MS, or higher degree, in a related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Knowledge of debugging and profiling deep neural networks on NVIDIA CUDA stack Experience in state of the art architectures for object detection and 3D perception Experience More ❯

architecture and scaling strategy About You Proficient hands-on experience training deep learning models, ideally at scale (multi-GPU, multi-node) Strong engineering skills in Python and PyTorch (or JAX); comfortable with distributed training setups Familiarity with dataset curation, versioning, and large-scale data management Hands-on experience running and analysing hyperparameter sweeps Deep curiosity about what’s happening at More ❯

the best researchers in the field. You May Be a Good Fit If You Have Extremely strong software engineering skills. Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR. Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray). Experience using large-scale distributed training strategies. Hands on experience on training large More ❯

modern AI/ML methods (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-driven workflows and tools. Passion More ❯

modern AI/ML methods (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-driven workflows and tools.- Passion More ❯

QUALIFICATIONS - Experience with building, validating and deploying GenAI models and applications on cloud infrastructure. - Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. - Knowledge of MLOps tools and workflows for model development, validation, and deployment. - High level of communicating effectively across internal and external organizations, with both technical and non-technical More ❯

QUALIFICATIONS - Experience with building, validating and deploying GenAI models and applications on cloud infrastructure. - Sound knowledge of machine learning fundamentals, with working knowledge of frameworks such as Pytorch, TensorFlow, JAX or MXNet. - Knowledge of MLOps tools and workflows for model development, validation, and deployment. - High level of communicating effectively across internal and external organizations, with both technical and non-technical More ❯

Life at UiPath The people at UiPath believe in the transformative power of automation to change how the world works. We're committed to creating category-leading enterprise software that unleashes that power. To make that happen, we need people More ❯

to study their performance. Proficiency in Python software development, demonstrated by published software projects (e.g., GitHub). Experience in developing and implementing deep learning systems (e.g., in PyTorch or JAX). Preferred: Experience working with graph data, reinforcement learning, and/or generative models. Experience with molecular data in real-world drug discovery. Track record of publications in ML conferences More ❯

systems (AI) & reinforcement learning) Comfort 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 a fast-paced environment—ideal More ❯

systems (AI) & reinforcement learning) Comfort 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 a fast-paced environment—ideal More ❯

systems (AI) & reinforcement learning) Comfort 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 a fast-paced environment—ideal More ❯

proficient software engineer in a least one programming language (Python or other e.g. Rust, Go, Java) • You have, hands-on experience with AI frameworks and tools (e.g., TensorFlow, PyTorch, Jax) • You have high technical engineering competence. This means being able to design complex software and make it usable in production • You have a high scientific track record in a field More ❯

experience: B.S./M.S. in Computer Science or related quantitative field with 5+ years of relevant experience. In addition, any of the following would be an advantage: Experience with JAX, PyTorch, or similar machine learning platforms, and on internal ML frameworks such as Gemax Scale/Prod and the Evergreen ecosystem. Python readability and/or strong artifacts in building More ❯

in the field. You may be a good fit if you have: Extremely strong software engineering skills. Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR. Experience writing kernels for GPUs using CUDA. Experience using large-scale distributed training strategies. Familiarity with autoregressive sequence models, such as Transformers. Bonus : paper at top More ❯

engineering skills, with a proven track record of building robust and scalable systems. Have a strong command of Python and are well-versed in popular deep learning frameworks like JAX, PyTorch, and TensorFlow, with an understanding of their multimodal capabilities. Knowledge of distributed training strategies, especially for large-scale multimodal models. Familiarity with autoregressive models, particularly their application in multimodal More ❯

million in seed funding Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques such as quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning frameworks like PyTorch. Experience building Deep Reinforcement Learning models. Experience with robotics and/or simulation. If you More ❯

years of ML experience, with a strong record of transitioning research into production. Proven team leadership in a technical or research setting. Expert in Python and PyTorch; JAX is a plus. Strong foundation in data preparation, feature engineering, and model evaluation. Excellent communicator with the ability to adapt messaging for different audiences. Strategic mindset and leadership potential at the executive More ❯

the best researchers in the field. You may be a good fit if you have: Extremely strong software engineering skills. Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR. Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray). Experience using large-scale distributed training strategies. Hands on experience on training large More ❯

the best researchers in the field. You may be a good fit if you have: Extremely strong software engineering skills. Proficiency in Python and related ML frameworks such as JAX, Pytorch and XLA/MLIR. Experience with distributed training infrastructures (Kubernetes, Slurm) and associated frameworks (Ray). Experience using large-scale distributed training strategies. Hands on experience on training large More ❯

Expertise and Requirements With expertise in production environments performing tasks like ML deployment using technologies such as PyTorch, TensorFlow, and JAX. Also, proficiency in frontend skills using React, TypeScript, or similar technologies. Experience in big data, real-time analytics, and More ❯