1 to 25 of 41 Remote Permanent JAX Jobs

AI Research Internship - Students Pursing PhD

Hiring Organisation: MediaTek
Location: London, England, United Kingdom

clearly Nice-to-have Experience in optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Previous research publications or submitted papers Why Join Us Gain hands-on experience in state-of-the-art AI research with real-world applications Collaborate ...

Artificial Intelligence Researcher

Hiring Organisation: microTECH Global LTD
Location: City of London, London, United Kingdom

modelling, RLHF). Hands-on experience training/fine-tuning generative models (LLMs, diffusion, transformers, GANs). Strong knowledge of deep learning frameworks (PyTorch, JAX, TensorFlow). Proficiency in Python and standard ML libraries. Solid foundations in probability, optimisation, and statistics. Experience working with large-scale distributed training on GPUs ...

Artificial Intelligence Researcher

Hiring Organisation: microTECH Global LTD
Location: London Area, United Kingdom

Deep Learning Researcher

Hiring Organisation: Microtech Global Ltd
Location: Cambridge, Cambridgeshire, East Anglia, United Kingdom
Employment Type: Permanent, Work From Home

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 ...

Founding Machine Learning Engineer

Hiring Organisation: Bjak
Location: Dublin, Ireland
Employment Type: Permanent
Salary: EUR 80,000 - 120,000 Annual

Strong background in deep learning and transformer architectures Hands-on experience training or fine-tuning large models (LLMs or vision models) Proficiency with PyTorch, JAX, or TensorFlow Experience with distributed training frameworks (DeepSpeed, FSDP, Megatron, ZeRO, Ray) Strong software engineering skills - writing robust, production-grade systems Experience with GPU optimization ...

Senior Researcher (Multimodal AI & Generative Models)

Hiring Organisation: Intelix.AI
Location: London, UK
Employment Type: Full-time

models and generative AI Proven experience with text-to-image generation, multimodal AI Large-scale model training experience (distributed systems, GPUs) Proficiency in PyTorch, JAX, or similar frameworks Enterprise AI deployment experience Safety and alignment research background Multilingual/multicultural AI expertise Highly Valued: A Background at top-tier ...