12 of 12 JAX Jobs

forests, and ensemble methods Experience with generative models (GANs, VAEs, Diffusion models) Programming & Tools : Expert-level Python programming Proficiency with ML frameworks (PyTorch, TensorFlow, JAX) Experience with scikit-learn, XGBoost, LightGBM Strong software engineering practices and clean code principles Data & Computing : Experience with big data technologies (Spark, Dask ...

Vision-Language Models (VLMs) Strong foundation in machine learning and deep learning fundamentals Excellent coding skills in Python, with experience in PyTorch, TensorFlow, or JAX Demonstrated ability to work collaboratively in cross-functional teams and communicate complex ideas to both technical and non-technical stakeholders Experience in applying responsible ...

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

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

Minimum Requirements: In-depth knowledge of Python for high-performance, data-intensive applications. Proficiency with at least one modern deep learning framework (e.g., PyTorch, Jax, TensorFlow Expertise in one or more of the following areas: computer vision, large language models, recommender systems, or operations research. Foundational knowledge of statistics ...

front-end visualization systems and dashboards. Familiarity with ML systems evaluations. Proficiency in programming languages such as Python and ML frameworks (e.g., PyTorch, TensorFlow, JAX). Excellent communication skills to collaborate effectively with cross-functional teams and present findings. One or more papers at top-tier venues (such as NeurIPS ...

other than Triton. The Team The Frameworks team ensures Graphcore hardware works seamlessly with the tools that ML engineers and researchers love – Pytorch, Triton, Jax and TensorFlow. We’re a talented and diverse team of engineers and we foster a culture of collaboration, openness and learning. All our software teams ...

At Plumerai, we make it easy and affordable for developers to add highly accurate AI to their embedded devices and thereby enable them to create amazing new products. We combine our on-device Tiny AI ...

Overview At Microsoft Research AI for Science we believe deep learning has the potential to transform scientific modelling and discovery crucial for solving the most pressing problems facing society, including sustainable materials and discovery of ...

Join Tether and Shape the Future of Digital Finance At Tether, we're not just building products, we're pioneering a global financial revolution. Our cutting edge solutions empower businesses-from exchanges and wallets to ...

*Must be Sole British National for this role* This is an exciting opportunity to join a cutting-edge technology consultancy working at the forefront of innovation in defence and homeland security. As a Machine Learning ...

approaches, and lessons learned You May Be a Good Fit If You Have hands-on experience training large language models, or deep expertise with JAX, TPU, PyTorch, or large-scale distributed systems Genuinely enjoy both research and engineering work—you'd describe your ideal split as roughly 50/… Care about the societal impacts of AI and responsible scaling Strong Candidates May Also Have Previous experience training LLM’s or working extensively with JAX/TPU, PyTorch, or other ML frameworks at scale Contributed to open-source LLM frameworks (e.g., open_lm, llm-foundry, mesh-transformer-jax) Published research ...