1 to 25 of 75 Google JAX Jobs

scaler platforms, e.g., AWS, Azure, GCP); building machine learning models and pipelines in Python, using common libraries and frameworks (e.g., NumPy, SciPy, Pandas, PyTorch, JAX), especially including deep learning applications; C/C++ for computer vision, geometry processing, or scientific computing; software engineering concepts and best practices (e.g., versioning ...

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

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

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

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

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

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

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

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

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

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

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

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

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

engineers through technical leadership, code reviews and best practices. Building systems that balance performance, latency, reliability and scalability. Technology Python PyTorch and/or JAX GPU-based model training and inference Modern machine learning infrastructure What We're Looking For Commercial experience building and deploying machine learning models into ...

help shape engineering best practices. Balance research ambitions with practical engineering trade-offs to deliver value quickly. Technology Python PyTorch and/or JAX Large-scale GPU training and inference infrastructure Modern ML deployment pipelines Ideal Experience Proven experience delivering machine learning systems into production. Strong understanding of modern ...