1 to 25 of 83 Google JAX Jobs

keywords. Useful experience Experience with some of the following would be valuable, but we do not expect you to know everything: PyTorch, TensorFlow or JAX Hugging Face LLM APIs and open-source models RAG and vector databases Apache Airflow AWS S3 ECS/EKS Redshift Docker Kubernetes GitHub Actions ...

Senior) or 8+ years (Staff) of hands‐on applied ML experience, including production‐grade work. Deep familiarity with modern deep learning stack (PyTorch or JAX), and breadth across at least two of: generative models (diffusion, autoregressive, flow‐based), graph and equivariant networks, vision (CNNs, ViTs, multimodal LLMs), search ...

track record of building and maintaining production‐quality systems. Expert‐level Python experience with numerical, statistical, or ML libraries (e.g. NumPy, pandas, PyTorch, JAX, TensorFlow). Deep understanding of statistical modelling, optimisation, or machine‐learning techniques. Experience building and operating backend systems and data pipelines in a cloud environment ...

track record of building and maintaining production‐quality systems. Expert‐level Python experience with numerical, statistical, or ML libraries (e.g. NumPy, pandas, PyTorch, JAX, TensorFlow). Deep understanding of statistical modelling, optimisation, or machine‐learning techniques. Experience building and operating backend systems and data pipelines in a cloud environment ...

/ML: Expertise in one or more Public Cloud offerings ( AWS, GCP, Azure, OCI ) and familiarity with AI/ML frameworks ( PyTorch, Tensorflow, JAX ) and MLOps. High-Throughput I/O: Experience with cutting-edge I/O architectures like DAOS or SPDK . Networking Foundations: Background in RDMA ...

language understanding. The ability to write clean and maintainable code in Python, Rust, C++ or Java Experience with frameworks such as PyTorch, Tensorflow or JAX Strong written and verbal communication skills to document your work effectively and present it to relevant stakeholders and users A passion for identifying ...

computer science, or a related field. Deep practical experience in Python and at least one modern machine learning framework such as PyTorch, TensorFlow, or JAX, experience working with large compute clusters and ML infrastructure is a plus. A track record of leading self-directed research projects that go well ...

computer science, or a related field. Deep practical experience in Python and at least one modern machine learning framework such as PyTorch, TensorFlow, or JAX, experience working with large compute clusters and ML infrastructure is a plus. A track record of leading self-directed research projects that go well ...

physics, computer science, or a related field.Deep practical experience in Python and at least one modern machine learning framework such as PyTorch, TensorFlow, or JAX, experience working with large compute clusters and ML infrastructure is a plus.A track record of leading self-directed research projects that go well beyond ...

good fit if you:Are proficient in Python and async/concurrent programming with frameworks like TrioHave experience with machine learning frameworks (PyTorch, TensorFlow, JAX)Have industry experience in machine learning researchCan balance research exploration with engineering implementationEnjoy pair programming (we love to pair!)Care about code quality, testing ...

Mathematics, etc.) or equivalent practical experience.ML Fundamentals: Deep understanding of machine learning principles and techniques.Framework Expertise: Strong proficiency in deep learning frameworks such as JAX or PyTorch.Modern Architectures: Hands-on experience building and working with modern model architectures (e.g., Transformers, GNNs, Diffusion Models).Full ML Lifecycle: Experience taking models ...

Infrastructure as Code, and DevOps practices Production support, monitoring, observability, and incident management Experience with AI/ML frameworks such as PyTorch, TensorFlow, or JAX Exposure to LLMs, RAG, Agentic AI, model serving, or MLOps environments What's on Offer? Work on innovative AI and Agentic AI solutions Hybrid ...

Agents, reasoning models, and RAG Have high-level familiarity with the architecture and operation of large language models Have personally implemented models in PyTorch, Jax, or TensorFlow Have a strong foundation in data structures, algorithms, and software engineering principles Exhibit excellent problem-solving and analytical skills with a proactive ...

applied AI research. Commercial or research experience implementing machine learning algorithms. Strong programming skills in Python; C++ experience is advantageous. Experience with PyTorch, TensorFlow, JAX, or similar deep learning frameworks. Why Apply? Join a world-class AI research team working on cutting-edge machine learning technologies. Play ...

related frameworks• Experience on AI/ML model implementations using scikit learn, Tensor flow etc.,• Experience with Chat, IVR, Banking will be plus• Google Cloud (GCP) knowledgeAbout NTT DATANTT DATA is a $30 billion business and technology services leader, serving 75% of the Fortune Global 100. We are committed ...

Java, and their associated security considerations.Coding experience with AI & ML, deep learning and data manipulation techniques, libraries, and frameworks such as TensorFlow, PyTorch, Jax, and scikit-learn.Familiarity with current and emerging regulations and standards, such as the EU AI Act, the Digital Services Act, the Digital Markets Act, GDPR ...

Python, C++, or similar Experience with optimization, reinforcement learning, and/or large language models (LLMs) Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience with scalable system design and deployment in research or production environments Exposure to chip design or applied AI research TPBN1_UKTJ ...

equivalent practical experience. Strong Python programming and software‐engineering skills. Experience developing machine‐learning, generative AI or LLM applications. Practical experience with PyTorch, JAX, TensorFlow or a similar framework. Understanding of agentic AI concepts such as tool calling, planning, memory and workflow orchestration. Knowledge of reinforcement learning, preference optimisation ...

related discipline, or equivalent practical experience.Strong Python programming and software-engineering skills.Experience developing machine-learning, generative AI or LLM applications.Practical experience with PyTorch, JAX, TensorFlow or a similar framework.Understanding of agentic AI concepts such as tool calling, planning, memory and workflow orchestration.Knowledge of reinforcement learning, preference optimisation or sequential ...

prototypes to reliable deployment; Have experience partnering closely with researchers throughout implementation; Be proficient in Python and deep learning frameworks such as PyTorch or JAX; Have strong software engineering fundamentals, including clean, testable, well‐documented and maintainable code, version control and code reviews; Have experience with distributed training frameworks ...

reliable deployment Experience working in close partnership with researchers throughout the implementation process Proficiency in Python and deep learning frameworks such as PyTorch or JAX) Strong software engineering fundamentals - writing clean, testable, well-documented and maintainable code, version control, code reviews Experience with distributed training frameworks such as PyTorch ...

strong programming skills in Python , Rust, or similar; and strong software engineering fundamentals building performant and reliable systems. Proficient in deep learning frameworks (Pytorch, JAX, TensorFlow). You can work on different layers of the stack from low-level training backends, data ingestion to ML/RL algorithmic design ...

record of shipping products and features inside a fast-moving environment Strong coding skills in Python and familiarity with ML frameworks like PyTorch or JAX Experience training, fine-tuning, and evaluating large language models Deep familiarity with large language models and web agents, how they work, and how they ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...

expertise. Creative thinkers who are driven, self-motivated, and eager to solve challenging problems Proficiency in Python and/or C++ Proficiency in Pytorch, JAX, Tensorflow or other DL library. Ability to thrive in a collaborative, team-oriented environment Expertise in GPU or Accelerator programming (CUDA, Triton, SYCL, ROCm ...