JAX Jobs

17 of 17 JAX Jobs

Research Scientist - Salesforce AI Research

California, United States
Salesforce.com, Inc
programming languages, including but not limited to C/C++ or Python. Experience with one or more deep learning libraries and platforms (e.g., TensorFlow, Jax, or PyTorch). Proven ability to implement, operate, and deliver results via innovation at a large scale. Large Language Models (LLMs), LLM evaluation and benchmarking more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Research Scientist - Salesforce AI Research

Palo Alto, California, United States
Salesforce, Inc
programming languages, including but not limited to C/C++ or Python. Experience with one or more deep learning libraries and platforms (e.g., TensorFlow, Jax, or PyTorch). Proven ability to implement, operate, and deliver results via innovation at a large scale. Large Language Models (LLMs), LLM evaluation and benchmarking more »
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer (Hiring Immediately)

London, UK
Finster AI
degree in Computer Science or relevant fields or relevant experience Have strong programming experience and familiarity with Python based deep learning frameworks like Pytorch, JAX, Tensorflow Have strong familiarity and knowledge of machine learning concepts Have familiarity with ML related maths like linear algebra, probability and statistics, multivariate calculus Have more »
Employment Type: Full-time
Posted:

AI Engineer

London Area, United Kingdom
Finster AI
degree in Computer Science or relevant fields or relevant experience Have strong programming experience and familiarity with Python based deep learning frameworks like Pytorch, JAX, Tensorflow Have strong familiarity and knowledge of machine learning concepts Have familiarity with ML related maths like linear algebra, probability and statistics, multivariate calculus Have more »
Posted:

Data Scientist / AI Engineer (TensorFlow, PyTorch)

Norwich, Norfolk, United Kingdom
Chameleon IT Solutions
language processing, including hands-on experience with transformer architectures (e.g., GPT, BERT, T5). Expertise in deep learning frameworks such as TensorFlow , PyTorch , or JAX . Strong knowledge of machine learning algorithms and their application to NLP tasks. Experience with cloud platforms such as AWS, GCP, or Azure for model more »
Employment Type: Permanent
Salary: £60000 - £68000/annum Bonus + Benefits
Posted:

Staff Cloud Solutions Architect, Rapid Innovation with Security Clearance

Reston, Virginia, United States
Google, Inc
years of experience in a customer-facing role. Experience designing and deploying with one or more from the following ML frameworks: TensorFlow, PyTorch, JAX, Spark ML, etc. Knowledge of AI/ML related engineering from business use case to algorithm selection, training, tuning, inference, and training pipeline design and build. more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Cloud Solutions Architect, Rapid Innovation Team with Security Clearance

Reston, Virginia, United States
Google, Inc
sales organization or public procurement organization. Experience designing and deploying with one or more from the following Machine Learning (ML) frameworks e.g., TensorFlow, PyTorch, JAX, Spark ML, etc. Experience with business use case to algorithm selection, training, tuning, inference, and training pipeline, designd and build. Knowledge of AI/ML. more »
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior Machine Learning Engineer

London Area, United Kingdom
Hybrid / WFH Options
Adamas Knight
models that enhance data-driven drug discovery, precision medicine, and computational biology. Create and optimize multi-modal machine learning algorithms, using frameworks such as JAX, PyTorch, and TensorFlow, with applications to vision and language understanding. Collaborate with interdisciplinary teams, including computational biologists, bioinformaticians, and data scientists, to apply machine learning more »
Posted:

Senior Machine Learning Engineer (Hiring Immediately)

London, UK
Hybrid / WFH Options
Adamas Knight
models that enhance data-driven drug discovery, precision medicine, and computational biology. Create and optimize multi-modal machine learning algorithms, using frameworks such as JAX, PyTorch, and TensorFlow, with applications to vision and language understanding. Collaborate with interdisciplinary teams, including computational biologists, bioinformaticians, and data scientists, to apply machine learning more »
Employment Type: Full-time
Posted:

Ai Software Engineer

United Kingdom
Hybrid / WFH Options
IC Resources
position? Strong C++ development skills Experience with parallel programming - CUDA, OpenCL or SYCL Strong understanding of computer architecture. GPU/CPU Experience with TensorFlow, JAX, NumPy or PyTorch If you are an Ai Software Engineer looking for a new opportunity within a rapidly growing, disruptive start up, please apply to more »
Posted:

Ai Software Engineer (Hiring Immediately)

United Kingdom, UK
Hybrid / WFH Options
IC Resources
position? Strong C++ development skills Experience with parallel programming - CUDA, OpenCL or SYCL Strong understanding of computer architecture. GPU/CPU Experience with TensorFlow, JAX, NumPy or PyTorch If you are an Ai Software Engineer looking for a new opportunity within a rapidly growing, disruptive start up, please apply to more »
Employment Type: Full-time
Posted:

Sr. Machine Learning Engineer (Hiring Immediately)

London, UK
Slingshot AI
have the right experience, if you’re ready to learn quickly on the job. 5+ years developing deep learning models in PyTorch, TensorFlow or JAX, include 3+ years in a production environment. Experience fine-tuning language models, like Llama. Experience with software engineering on a product, e.g. React, Swift, Kotlin more »
Employment Type: Full-time
Posted:

Sr. Machine Learning Engineer

London Area, United Kingdom
Slingshot AI
have the right experience, if you’re ready to learn quickly on the job. 5+ years developing deep learning models in PyTorch, TensorFlow or JAX, include 3+ years in a production environment. Experience fine-tuning language models, like Llama. Experience with software engineering on a product, e.g. React, Swift, Kotlin more »
Posted:

Data Scientist, AWS Generative AI Innovation Center

London, United Kingdom
Amazon
Data Scientist, AWS Generative AI Innovation Center Amazon launched the Generative AI Innovation Center (GenAIIC) in June 2023 to help AWS customers accelerate the use of generative AI to solve business and operational problems and promote innovation in their organization. more »
Employment Type: Permanent
Salary: GBP Annual
Posted:

Head of Research (LLMs)

London Area, United Kingdom
Amberes
a frontier lab Team leading experience Tangible experience in setting, executing & achieving research goals Experience working with multiple GPUs Experience with the PyTorch framework (JAX is not possible) Our client are looking for a truly exceptional Research leader who has made a significant contribution to their existing/previous organisation more »
Posted:

Machine Learning Engineer (RL)

London Area, United Kingdom
algo1
Machine Learning Engineer (Reinforcement Learning) About the job We are a VC-backed startup focused on behavioural AI, currently in stealth mode. We are building for retail with a focus on reinforcement learning, generative AI and recommendation systems. We are more »
Posted:

Staff Engineer, AI/ML Software Compiler

San Jose, California, United States
Hybrid / WFH Options
Samsung Semiconductor
Please Note: To provide the best candidate experience amidst our high application volumes, each candidate is limited to 10 applications across all open jobs within a 6-month period. Advancing the World's Technology Together Our technology solutions power the more »
Employment Type: Permanent
Salary: USD Annual
Posted:
JAX
25th Percentile
£86,525
Median
£100,000