Permanent JAX Job Vacancies

76 to 100 of 112 Permanent JAX Jobs

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:

Machine Learning Engineer

London Area, United Kingdom
Rain UK Labs
equivalent education and experience. Experience building and distributing software libraries (including developing code with unit tests and collaborating on Github). Experience with deep learning frameworks such as PyTorch, Jax or Tensorflow. Experience with implementing and training large models (e.g. ResNets, diffusion models, and transformers) on GPU clusters. Experience in distributed computing. Preferred qualifications: Understanding of deep learning models such More ❯
Posted:

Machine Learning Engineer

City of London, London, United Kingdom
Rain UK Labs
equivalent education and experience. Experience building and distributing software libraries (including developing code with unit tests and collaborating on Github). Experience with deep learning frameworks such as PyTorch, Jax or Tensorflow. Experience with implementing and training large models (e.g. ResNets, diffusion models, and transformers) on GPU clusters. Experience in distributed computing. Preferred qualifications: Understanding of deep learning models such More ❯
Posted:

(Senior) AI/ML Engineer -Embedded Performance (all)

Karlsruhe, Baden-Württemberg, Germany
Cinemo GmbH
different CPU (ARM/x86) and GPU architectures (Nvidia) Experience with state-of-the-art performance toolkits, software stacks and profiling tools (Cuda/CuDNN/TensorRT, Jit + Jax) Proficient knowledge of Python, C/C++, ARM/x86 assembly, PyTorch, TensorFlow, Keras Fundamental knowledge about parameter efficient fine-tuning techniques (e.g., LoRA, PEFT) Good written and verbal English More ❯
Employment Type: Permanent
Salary: EUR 85,000 - 100,000 Annual
Posted:

Director of AI and Research

City of London, London, United Kingdom
Hybrid / WFH Options
Formula Recruitment
related field. Demonstrated success shipping LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med-tech, pharma, fintech, etc.). Strong More ❯
Posted:

Director of AI and Research

London Area, United Kingdom
Hybrid / WFH Options
Formula Recruitment
related field. Demonstrated success shipping LLM-based products (GPT, Llama, Mistral, or similar) or conversational AI systems into production. Deep knowledge of modern ML/NLP frameworks (PyTorch, TensorFlow, JAX) and orchestration stacks (LangChain, LlamaIndex, vector databases). Proven track record driving AI governance, model risk management, and compliance within regulated sectors (healthcare, med-tech, pharma, fintech, etc.). Strong More ❯
Posted:

ML Ops Engineer, London

London, United Kingdom
RATA TRANSPORTATION LLC
experience with modern Machine Learning Ops lifecycle, tools, and frameworks Nice to have: Experience supporting machine learning research activities Familiarity with python-based ML frameworks like TensorFlow, PyTorch, or JAX Experience with Google Cloud Platform (GCP) Culture and values What does it take to be successful at IsoLabs? It's not about finding people who think and act in the More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Research Engineer, Chip Verification

Mountain View, California, United States
DeepMind
ML for optimization. In addition, the following would be an advantage: Experience with MLIR, and in particular dialects specialized for hardware synthesis, e.g., CIRCT, XLS, and similar. Experience with JAX, TensorFlow, PyTorch or similar. Experience or interest in hardware verification technologies, formal methods, theorem proving, and hardware testing methodologies. Self-directed engineer/research scientist who can drive new research More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Bio AI Research Scientist

London, United Kingdom
Prima Mente
ideally including transformers, diffusion models, or multimodal architectures. Experience applying ML to large multimodal biological datasets (genomics, transcriptomics, epigenomics, proteomics, single-cell data). Exceptional coding skills (Python, PyTorch, JAX, TensorFlow) and familiarity with scalable training frameworks. Experience managing large-scale distributed training across GPUs/TPUs. Background in generative AI, large language models, mechanistic interpretability, or reinforcement learning. Interview More ❯
Employment Type: Permanent
Salary: GBP Annual
Posted:

Model Optimization Engineer (PyTorch Infrastructure Development)

Cupertino, California, United States
Apple
or related discipline. 3+ years of industry and/or research experience Highly proficient in Python programming Proficiency in at least one ML authoring framework, such as PyTorch, TensorFlow, JAX, MLX Experience in the area of model compression and quantization techniques, specially in one of the optimization libraries for an ML framework (e.g. torch.ao). Ability to ramp up quickly More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

Senior Machine Learning Engineer

Belgium
Hybrid / WFH Options
IC Resources
such as quantisation, pruning, and distillation to improve performance. Contribute to deployment, testing, and continuous model improvement. Requirements: Strong experience building and deploying ML models using PyTorch or TensorFlow; JAX is a plus. Solid background in NLP, computer vision and/or speech processing. Knowledge of model optimisation and performance tuning. Familiarity with distributed or edge AI environments. Passion for More ❯
Employment Type: Permanent
Salary: EUR Annual
Posted:

Machine Learning Engineer

Eindhoven, Noord-Brabant, Netherlands
Hybrid / WFH Options
IC Resources
such as quantisation, pruning, and distillation to improve performance. Contribute to deployment, testing, and continuous model improvement. Requirements: Strong experience building and deploying ML models using PyTorch or TensorFlow; JAX is a plus. Solid background in NLP, computer vision and/or speech processing. Knowledge of model optimisation and performance tuning. Familiarity with distributed or edge AI environments. Passion for More ❯
Employment Type: Permanent
Salary: EUR Annual
Posted:

Machine Learning Applied Researcher

London Area, United Kingdom
Harnham
Experience across 3D graphics, rendering, or materials/textures is required Experience implementing and training diffusion or generative image models Strong Python skills and familiarity with frameworks like PyTorch, JAX, or TensorFlow Understanding of transfer learning, computer vision, and image processing techniques Ability to translate complex research into scalable, production-ready solutions Why join this team Work with a unique More ❯
Posted:

Machine Learning Applied Researcher

City of London, London, United Kingdom
Harnham
Experience across 3D graphics, rendering, or materials/textures is required Experience implementing and training diffusion or generative image models Strong Python skills and familiarity with frameworks like PyTorch, JAX, or TensorFlow Understanding of transfer learning, computer vision, and image processing techniques Ability to translate complex research into scalable, production-ready solutions Why join this team Work with a unique More ❯
Posted:

AI Engineer - Generative Geometry for Hardware Design

United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

Sonoma, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

Fremont, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

Sunnyvale, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

Hayward, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

Alameda, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

Santa Rosa, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

San Mateo, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

San Jose, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

San Francisco, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:

AI Engineer - Generative Geometry for Hardware Design

Santa Clara, California, United States
Vinci4D.ai
mesh generation, implicit surfaces, or neural fields (e.g., NeRF, SDF, DeepSDF, Occupancy Networks). Experience with the related technologies, libraries, and languages: Python, C++, PyTorch (3D)/TensorFlow/JAX; plus to have GPU programming Experience with diffusion models for 3D generation Startup experience is a strong advantage. Understanding of geometry representations (mesh, voxel, point cloud, NURBS, parametric surfaces). More ❯
Employment Type: Permanent
Salary: USD Annual
Posted:
JAX
10th Percentile
£68,100
25th Percentile
£77,250
Median
£95,000
75th Percentile
£97,500