1 to 25 of 90 Permanent JAX Jobs

language) and in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling workflows. ML Ops & deployment: Familiarity More ❯

language) and in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling workflows. ML Ops & deployment: Familiarity More ❯

language) and in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling workflows. ML Ops & deployment: Familiarity More ❯

language) and in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling workflows. ML Ops & deployment: Familiarity More ❯

language) and in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modeling workflows. ML Ops & deployment: Familiarity More ❯

projects including training, fine-tuning, or production deployment Strong track record in scalable AI system development Technical Expertise Expert-level Python programming with deep knowledge of PyTorch, TensorFlow, or JAX Proven experience with distributed computing environments and modern MLOps workflows Practical knowledge of LLM toolkits (Hugging Face ecosystem, training frameworks) Strong problem-solving abilities for complex model performance and infrastructure More ❯

projects including training, fine-tuning, or production deployment Strong track record in scalable AI system development Technical Expertise Expert-level Python programming with deep knowledge of PyTorch, TensorFlow, or JAX Proven experience with distributed computing environments and modern MLOps workflows Practical knowledge of LLM toolkits (Hugging Face ecosystem, training frameworks) Strong problem-solving abilities for complex model performance and infrastructure More ❯

projects including training, fine-tuning, or production deployment Strong track record in scalable AI system development Technical Expertise Expert-level Python programming with deep knowledge of PyTorch, TensorFlow, or JAX Proven experience with distributed computing environments and modern MLOps workflows Practical knowledge of LLM toolkits (Hugging Face ecosystem, training frameworks) Strong problem-solving abilities for complex model performance and infrastructure More ❯

projects including training, fine-tuning, or production deployment Strong track record in scalable AI system development Technical Expertise Expert-level Python programming with deep knowledge of PyTorch, TensorFlow, or JAX Proven experience with distributed computing environments and modern MLOps workflows Practical knowledge of LLM toolkits (Hugging Face ecosystem, training frameworks) Strong problem-solving abilities for complex model performance and infrastructure More ❯

projects including training, fine-tuning, or production deployment Strong track record in scalable AI system development Technical Expertise Expert-level Python programming with deep knowledge of PyTorch, TensorFlow, or JAX Proven experience with distributed computing environments and modern MLOps workflows Practical knowledge of LLM toolkits (Hugging Face ecosystem, training frameworks) Strong problem-solving abilities for complex model performance and infrastructure More ❯

projects including training, fine-tuning, or production deployment Strong track record in scalable AI system development Technical Expertise Expert-level Python programming with deep knowledge of PyTorch, TensorFlow, or JAX Proven experience with distributed computing environments and modern MLOps workflows Practical knowledge of LLM toolkits (Hugging Face ecosystem, training frameworks) Strong problem-solving abilities for complex model performance and infrastructure More ❯

expertise: Skilled in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling workflows. Model development & deployment: Familiarity More ❯

expertise: Skilled in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling workflows. Model development & deployment: Familiarity More ❯

expertise: Skilled in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling workflows. Model development & deployment: Familiarity More ❯

expertise: Skilled in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling workflows. Model development & deployment: Familiarity More ❯

expertise: Skilled in building data pipelines and ML models using modern libraries across multiple domains: Data science stack: NumPy, pandas/polars, scikit-learn, XGBoost, LightGBM Deep learning: PyTorch, JAX Statistical programming: NumPyro, PyMC Data skills: Proficient in SQL, with the ability to write efficient, maintainable queries and manage data pipelines for analytics and modelling workflows. Model development & deployment: Familiarity 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 ❯

Flask) Frontend (e.g. React) and backend (e.g. Node.js) development Machine Learning & Artificial Intelligence Machine Learning (General concepts, model development, experimentation, training, evaluation) Deep Learning (General, frameworks like TensorFlow, PyTorch, JAX, Keras, neural networks, CNNs, RNNs, transformer architecture, LSTM) Natural Language Processing (NLP) Reinforcement Learning (e.g., PPO, Q-learning, policy gradients, A2C, DQN, AlphaZero) Computer Vision (e.g., image processing, analysis, instance More ❯

ability to present complex ideas 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 with a team of world More ❯

ability to present complex ideas 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 with a team of world More ❯

ability to present complex ideas 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 with a team of world More ❯