Permanent JAX Job Vacancies

26 to 46 of 46 Permanent JAX Jobs

Senior Member of Technical Staff, Multimodal AI

London, United Kingdom
Cohere
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 ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

AI Engineer / Data Scientist

Musselburgh, Midlothian, United Kingdom
Daxtra Technologies Ltd
About Us: Daxtra is at the forefront of AI-driven innovation, leveraging cutting-edge technologies to develop intelligent solutions that enhance business outcomes. We are looking for an AI Engineer to work closely with our AI/ML team, contributing More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Member of Technical Staff, Post-Training

London, United Kingdom
Cohere
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 ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer - NYC- Leading Global Prop Trading Firm - Oxford Knight

New York, United States
Oxford Knight
techniques and linear algebra. Eagerness to stay updated on cutting-edge ML advancements. Ability to create and maintain organized, reproducible research code. Expertise in ML frameworks, particularly PyTorch, or Jax, TensorFlow or similar. Innovative mindset, with a focus on questioning and improving methods. NB: Please only apply if you meet the above criteria. Benefits and Incentives Annual bonus. Health insurance More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Member of Technical Staff, Training Performance Engineer

London, United Kingdom
Cohere
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 writing kernels for GPUs using CUDA, triton, etc Experience using large-scale distributed training strategies. Familiarity with autoregressive sequence models, such as Transformers. Bonus More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Staff Machine Learning Engineer - Modeling

Cambridge, Cambridgeshire, United Kingdom
Second Renaissance
inference, causal reasoning & inference, transfer & multi-task learning, diffusion models, graph neural networks, active learning; Experience writing production-quality code with modern machine learning frameworks such as PyTorch, TensorFlow, JAX, or similar; Experience with multi-GPU and distributed training at scale; A team player who thrives in collaborative environments and is committed to enabling colleagues to reach their full potential … physics, or computational biology, or equivalent experience; Experience in developing machine learning models; Very strong programming skills, including experience with Python and deep learning libraries such as PyTorch or JAX; Experience in large-scale distributed optimization of machine learning models across multiple GPUs and nodes. Preferred Qualifications - Familiarity with biological data formats, concepts, and computational models; Experience in cell health More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer (London)

London, UK
Cerebras
As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯
Employment Type: Full-time
Posted:

Machine Learning Engineer (London)

London, UK
Graphcore
Join to apply for the Machine Learning Engineer role at Graphcore Join to apply for the Machine Learning Engineer role at Graphcore Get AI-powered advice on this job and more exclusive features. As a Machine Learning Engineer in the More ❯
Employment Type: Full-time
Posted:

Machine Learning Engineer (Large Systems)

London, United Kingdom
Cerebras
As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer (Large Systems)

Cambridge, Cambridgeshire, United Kingdom
Cerebras
As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Research Scientist

London, United Kingdom
Cerebras
Job Summary As a researcher at Graphcore, you will contribute to the advancement of AI research, investigating new ideas that push the limits on important AI/ML problems. Specialised hardware has been the key driver of the progress of More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Engineer (Large Systems)

Bristol, Gloucestershire, United Kingdom
Cerebras
As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Machine Learning Scientist - Computational Biology (Multiple Levels)

Cambridge, Cambridgeshire, United Kingdom
Deep Genomics Inc
About Us Deep Genomics is at the forefront of using artificial intelligence to transform drug discovery. Our proprietary AI platform decodes the complexity of genome biology to identify novel drug targets, mechanisms, and genetic medicines inaccessible through traditional methods. With More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Modelling Engineer (Mid-Staff)

London, United Kingdom
La Fosse Associates
Build the Future of AI with Optical Compute We're pioneering optical processors to train and run inference on large-scale AI models. Join a team of highly motivated and skilled engineers dedicated to rapid innovation and high-impact outcomes. More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Research Engineer, Machine Learning (Horizons) London, UK (London)

London, UK
Hybrid / WFH Options
Alcides Fonseca
Research Engineer, Machine Learning (Horizons) London, UK About Anthropic Anthropics mission is to create reliable, interpretable, and steerable AI systems. We want AI to be safe and beneficial for our users and for society as a whole. Our team is More ❯
Employment Type: Full-time
Posted:

Remote Machine Learning Compiler Engineer - Gensyn (London)

Surbiton, Greater London, UK
Hybrid / WFH Options
Blockchain Works
The world will be unrecognisable in 5 years. Machine learning models are driving our cars , testing our eyesight , detecting our cancer , giving sight to the blind , giving speech to the mute , and dictating what we consume, enjoy, and think . More ❯
Employment Type: Full-time
Posted:

Remote Machine Learning Compiler Engineer - Gensyn

London, United Kingdom
Hybrid / WFH Options
Blockchain Works
The world will be unrecognisable in 5 years. Machine learning models are driving our cars , testing our eyesight , detecting our cancer , giving sight to the blind , giving speech to the mute , and dictating what we consume, enjoy, and think . More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Applied Scientist

London, United Kingdom
Unlikely Artificial Intelligence Limited
At UnlikelyAI, we're looking for a Senior Applied Scientist to join our Applied Science team. This is a high-impact individual contributor position. You'll help drive the end-to-end lifecycle of projects: from identifying opportunities in literature More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Staff Applied Scientist

London, United Kingdom
Unlikely Artificial Intelligence Limited
At UnlikelyAI, we're looking for a visionary Staff Applied Scientist to play a key leadership role in our Applied Science team and across the company. This is a high-impact individual contributor position with mentoring and coaching responsibilities - ideal More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Staff Research Engineer, Applied ML

London, United Kingdom
Google Inc
years of experience in applied machine learning, including leading projects from research to production. Experience building and leading engineering or research teams. Experience in Python and ML frameworks like JAX, TensorFlow, or PyTorch. One or more publications in top-tier ML/AI conferences or journals (e.g., NeurIPS, ICML, ICLR, CVPR). Preferred qualifications: Master's degree in Computer Science … in Google Research, DeepMind, and product areas to identify and pursue high-impact opportunities. Utilize and advance cutting-edge techniques on Google's infrastructure. Your work will involve Python, JAX/TensorFlow, and focus on areas like parameter-efficient tuning (PET), multimodal modeling, and agentic systems using models like Gemini. Research and product by translating complex technical concepts into actionable More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Senior Applied Scientist

London, United Kingdom
Hybrid / WFH Options
UiPath
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 ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:
JAX
10th Percentile
£61,000
25th Percentile
£65,250
Median
£85,000
75th Percentile
£97,500