1 to 25 of 94 Google JAX Jobs

track record of developing data-driven machine learning systems or products Proficiency in Python and relevant deep learning frameworks - both training (e.g. PyTorch, Tensorflow, JAX) and serving (e.g., Hugging Face TGI/Transformers/Adapters/outlines, vLLM) Experience with cloud deployment of ML systems (e.g., AWS, GCP, Azure ...

Python skills and comfort with data-science libraries (NumPy, pandas, scikit-learn) Foundational knowledge of AI/ML concepts and frameworks (PyTorch, TensorFlow, or JAX) Exposure to reinforcement learning or similar methods (deep Q-learning, PPO) through coursework, labs, or projects Understanding of model training life cycle: data prep ...

Master's degree in Computer Science, Machine Learning, Computational Biology, or related field 2+ years of hands-on experience with PyTorch and/or JAX for deep learning applications Strong proficiency in Python scientific computing stack (NumPy, Pandas, scikit-learn) Experience with distributed computing and GPU optimization techniques Familiarity ...

maintainer of any open-source projects Familiarity with containers and container orchestration systems Expertise in deep learning frameworks and tensor libraries like PyTorch, Tensorflow, JAX/FLAX Suggested Skills ML Algorithm Development Experience in Machine Learning and Deep Learning Experience in Information retrieval/recommendation systems/distributed serving ...

similar field. * Experience building, deploying and maintaining NLP/Deep Learning Systems. * Extensive experience in at least one Deep Learning framework (PyTorch, TensorFlow, JAX). * Well versed in the scientific Python ecosystem (NumPy, Scikit-Learn, Pandas etc.). * Strong Data Engineering underpinnings and an ability to work with ...

machine learning models or LLMs Experience scaling and optimizing the training and serving of machine learning models Experience with machine learning libraries TensorFlow, PyTorch, JAX or Keras Experience with cloud computing platforms like AWS Background in math, statistics, or numerical computation Significant contributions to open-source projects Note: Level ...

critical domains like fraud detection, content moderation, or risk assessment Are proficient in Python and have experience with ML frameworks like PyTorch, TensorFlow, or JAX Have hands-on experience with cloud platforms (AWS, GCP) and container orchestration (Kubernetes) Understand distributed systems principles and have built systems that handle high ...

building and shipping robust ML/AI models for large-scale, user-facing or data-driven products. Deep expertise in deep learning (PyTorch, TensorFlow, JAX), LLMs, information retrieval, content summarization, recommendation systems, NLP, and/or ranking. Strong software engineering skills (Python, production-quality codebases, collaborative development). ...

engineering skills in Python or C++. Strong analytical skills in linear algebra, geometry, and probability. Experience with a deep learning framework, such as PyTorch, JAX, TensorFlow. Extra Credit Relevant industry experience with low-latency systems (prior work on self-driving vehicles, autonomy, and/or robotics projects). Experience ...

millions of users globally, in a massive and growing market opportunity. Yupp's founding team is highly experienced and comes from companies like Twitter, Google, Coinbase, Microsoft and Paypal. This team is one of the smartest, most fun, cracked top talent you will ever work with. Our work culture ...

advancements into scalable, real-world solutions. Ideally, you'd have: Practical experience working with LLMs, with proficiency in frameworks like PyTorch, JAX, or TensorFlow. You should also be skilled at rapidly interpreting research literature and turning new ideas into working prototypes. A track record of published research ...

/ML engineering, with a strong track record of delivering production-grade AI systems. Deep expertise in Python and modern ML frameworks (PyTorch, TensorFlow, JAX). Extensive hands-on experience working with large language models, generative AI architectures, and related tooling. Strong knowledge of model serving, orchestration, distributed training ...

learning algorithms (e.g., Transformer/VLM/LLM, CNNs, residual networks, GAN, clustering, sequence models) and frameworks/libraries (e.g., Tensorflow, PyTorch, Scikit-learn, Jax). Excellent experience in deployment of machine learning models in production environments, particularly in video/media-focused applications. Excellent communication skills and ability ...

Learning, or related field. 5+ years of experience in applied machine learning, computer vision, or robotics. Strong background in deep learning frameworks (PyTorch, TensorFlow, JAX). Hands-on experience building and deploying ML models robotic manipulation for grasping, manipulation, or dexterous robotics. Expertise with large multimodal ML models (vision ...

Learning, or related field. 5+ years of experience in applied machine learning, computer vision, or robotics. Strong background in deep learning frameworks (PyTorch, TensorFlow, JAX). Hands-on experience building and deploying ML models robotic manipulation for grasping, manipulation, or dexterous robotics. Expertise with large multimodal ML models (vision ...

stack; Experience MS, or higher degree, in related field, or equivalent industry experience Professional experience with ML frameworks such as PyTorch, Jax or Tensorflow (PyTorch preferred) Experience with data sampling and data curation pipelines for autonomous driving datasets Experience in state of the art architectures ...

tracking, job monitoring, and logging usable by collaborators and researchers. • Infra Engineering - Write production-quality code and tests for ML infra in PyTorch or JAX; ensure reliability and maintainability at scale. Qualifications Must-Haves: • 5+ years of experience in ML systems, infra, or distributed training • Experience modifying distributed ...

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

Enthusiasm for R&D and pushing the boundaries of what AI can do for support Technologies We Use and Teach Python, PyTorch, TensorFlow, or JAX We're working to build a more inclusive economy where our customers have equal access to opportunity, and we strive to live by these ...

reimagining the healthcare technology stack to make the lives of clinicians better. Our tech stack includes GCP, Kubernetes, Vertex AI, Golang, Python, C++, PyTorch, JAX, Transformers, GPT, LLAMA, MoE, gRPC, Javascript. What We're Trying to Do (And Why We Need You) We're not here to tweak healthcare ...

record of published peer reviewed innovative AI/ML research Experience writing production-quality code with modern machine learning frameworks such as PyTorch, TensorFlow, JAX, or similar; Experience with multi-GPU and distributed training at scale ; Preferred Qualifications Familiarity with multimodal data integration, including early and/or late ...

operating, or maintaining feature and model stores. Previous contributions to open source ML Infrastructure projects or research publications in ML systems. Experience with Pytorch, JAX, or TensorFlow At Nuro, your base pay is one part of your total compensation package. For this position, the reasonably expected base pay range ...

Team The Frameworks team ensures Graphcore hardware works seamlessly with the tools that ML engineers and researchers love – Pytorch, Triton and in future, 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 ...

Team The Frameworks team ensures Graphcore hardware works seamlessly with the tools that ML engineers and researchers love – Pytorch, Triton and in future, 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 ...

collaborating with other researchers to scale experimental ideas Strong candidates may also have: Experience with language model training infrastructure and distributed ML frameworks (PyTorch, JAX, etc.) Background in building infrastructure for AI research labs or large-scale ML organizations Knowledge of GPU/TPU architectures and language model inference ...