1 to 25 of 50 Permanent JAX Jobs

deep experience with the AWS ecosystem and services relevant to ML, such as SageMaker, EC2, S3, IAM, and VPC. You are familiar with deep-learning frameworks like PyTorch or JAX, and libraries from the AI ecosystem like Hugging Face Transformers. You have worked with the lifecycle of AI models, including deployment, hosting, and optimization for low-latency inference. You have More ❯

environments. Proficiency in production-grade programming languages such as Rust and C++. Decent understanding of distributed systems, cluster orchestration and management. Good knowledge of data science tools (e.g., PyTorch, Jax, Numpy) and programming languages such as Python. Experience with monitoring tools (Prometheus, Grafana). Experience working in Agile development teams with excellent collaboration skills. Grit in the face of technical More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 work environment Opportunities to publish More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 work environment Opportunities to publish More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 work environment Opportunities to publish More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 work environment Opportunities to publish More ❯

C++, and deep learning frameworks (e.g., PyTorch, TensorFlow) Solid grasp of ML techniques; independent and team-oriented mindse 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 work environment Opportunities to publish More ❯

Reinforcement Learning (policy optimisation, reward 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/TPUs. If this sounds More ❯

Reinforcement Learning (policy optimisation, reward 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/TPUs. If this sounds More ❯

Reinforcement Learning (policy optimisation, reward 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/TPUs. If this sounds More ❯

Reinforcement Learning (policy optimisation, reward 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/TPUs. If this sounds More ❯

Reinforcement Learning (policy optimisation, reward 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/TPUs. If this sounds More ❯

years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture. Experience using machine learning frameworks (e.g., Tensorflow, PyTorch, or Jax). Experience with machine learning model pretraining, fine-tuning, evaluation, and methodologies. Preferred qualifications: Master's degree or PhD in Computer Science or related technical field. Experience publishing in high-quality More ❯

learning and generative AI techniques, including implementing custom architectures, pretraining/fine-tuning generative models, building agentic workflows, and deploying solutions with real-world constraints. Familiarity with PyTorch, TensorFlow, JAX or similar frameworks Preferred Qualifications Experience in building AI agents (based on RL, LLMs, etc.) Experience in building multi-agent simulation Background in Large Language Models/NLP and Multi More ❯

of high-quality research, evidenced by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. Strong analytical and critical thinking skills More ❯

with existing data and machine learning systems Qualifications Strong understanding of machine learning principles, especially in the context of LLMs. Solid proficiency in Python, including machine learning packages like Jax/Tensorflow or PyTorch Experience with large-scale data infrastructure Strong verbal and written communication skills with the ability to work effectively across internal and external organisations and virtual teams. More ❯

Out from the Crowd: Advanced knowledge of LLMs, modern AI software architecture and cloud APIs Existing public facing technical content, forum contributions or open source projects Familiarity with PyTorch, JAX, vLLM or other training & inference frameworks #LI-Hybrid Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary More ❯

drive: Deep learning expertise with 2+ years in language generation. Experience with LLMs, neurosymbolic integration, and knowledge representation. Strong Python skills and experience with ML frameworks (PyTorch, HF, TensorFlow, JAX). Demonstrable track record of end-to-end applied research. Excellent communication skills; proven mentoring ability. Collaborative and proactive in fast-paced environments. Desirable: Experience with symbolic reasoning engines. Technical More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

looking for 3+ years developing deep learning systems with shipped models or published research. Hands-on with Vision Language Models, LLMs or generative models. Strong Python + PyTorch/JAX skills (debugging, profiling, shipping). Passion for bridging the gap between simulation and reality. This is one of the most exciting and technically challenging roles I’ve see with the More ❯

facing developer workflows. Experience developing for and on Arm architectures. Experience profiling or debugging ML workloads. Experience designing, training, optimizing, and deploying machine learning models using frameworks like PyTorch, JAX, LiteRT or HuggingFace Transformers. Experience collaborating closely with product or design teams. In Return: Our team offers an inclusive and highly collaborative environment, with work that spans a wide variety More ❯

reshaping the manufacturing world. ‍ About you A machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down and dirty with cleaning More ❯

reshaping the manufacturing world. ‍ About you A machine learning mage: You write clean, scalable, high-performance deep learning code. You take pride in your craft. Proficiency in PyTorch, Tensorflow, Jax, or equivalent is a must (ideally PyTorch). Knowledge on some of the latest good stuff from transformers to VLMs and NERFs. Happy to get down and dirty with cleaning More ❯