9 of 9 Google JAX Jobs

record of building and deploying LLM-powered applications in production , Deep experience applying large language models to complex, real-world problems , Strong proficiency in Python and ML frameworks (PyTorch, JAX, TensorFlow, Hugging Face Transformers) , Hands-on experience with advanced LLM techniques: RAG systems, chain-of-thought prompting, agentic tool use, and multi-agent coordination , Experience with distributed systems, cloud More ❯

to transition ML pipelines from POC to scalable production systems. Preferred tooling: Programming Languages - Python, Go, Java, C++ Infrastructure as Code (IaC) - Terraform ML Frameworks and Libraries - PyTorch, TensorFlow, JAX, HuggingFace, Scikit learn What we offer Competitive salary and comprehensive benefits package. Opportunities for professional growth within Nebius. Flexible working arrangements. A dynamic and collaborative work environment that values More ❯

Understanding of the ML lifecycle and state of the art ML Infrastructure technologies Experience with GPU and other type of HPC infrastructure Experience with training framework like PyTorch, Tensorflow, JAX Deep understanding of Ray and KubeRay Experience with ML Training/Inference profiling and optimization More ❯

also are on the look out for candidates who: Have deep familiarity with Python data ecosystem Understanding of Jupyter notebooks Exposure to machine learning libraries like PyTorch, XGBoost and JAX Understanding of crypto or traditional financial markets Strong API design and documentation skills What do you get in return? Up to 250k base (depending on experience) 3 days in More ❯

also are on the look out for candidates who: Have deep familiarity with Python data ecosystem Understanding of Jupyter notebooks Exposure to machine learning libraries like PyTorch, XGBoost and JAX Understanding of crypto or traditional financial markets Strong API design and documentation skills What do you get in return? Up to £250k base (depending on experience) 3 days in More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … 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 More ❯

At UnlikelyAI, we're looking for a Senior Applied Scientist to join our Applied Science team. This is a high-impact individual contributor position. You'll help drive the end-to-end lifecycle of projects: from identifying opportunities in literature More ❯

knowledge by documenting systems, debugging 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/50 rather than heavily weighted … craft as a research engineer 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 on model training More ❯

About Mistral At Mistral AI, we believe in the power of AI to simplify tasks, save time, and enhance learning and creativity. Our technology is designed to integrate seamlessly into daily working life. We democratize AI through high-performance, optimized More ❯