17 of 17 Google JAX Jobs

Gymnasium. Agentic AI Frameworks: LangChain, LlamaIndex, AutoGen, or CrewAI. API Development: FastAPI, Flask, or Node.js . Cloud Platforms: AWS (SageMaker, Lambda, Bedrock), Google Cloud AI, Azure AI. Containerization: Docker, Kubernetes. Version Control: Git, GitHub, or GitLab. Databases: SQL (PostgreSQL, MySQL), NoSQL (MongoDB, DynamoDB). DevOps Tools: CI/ ...

implementing machine learning and deep learning algorithms. Proficiency in programming languages such as Python or C++, and familiarity with ML frameworks such as TensorFlow, JAX or PyTorch. Strong communication skills, both written and verbal, with the ability to present complex technical information to a diverse audience. Excellent problem-solving ...

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

Neuron team is developing a deep learning compiler stack that takes neural network descriptions created in frameworks such as TensorFlow, PyTorch, and Jax, and converts them into code suitable for execution. As a Sr. ML Compiler Engineer, you will be responsible for identifying and designing solutions that enable efficient ...

function. Must-have experience 5 + years shipping ML systems (vision, NLP, or multimodal) that serve live traffic. Expert Python plus PyTorch or JAX; comfortable with distributed training and experiment tracking. Track record of taking a research spike to a monitored, fault-tolerant, low-latency service. Deep understanding ...

Good Fit If You Have Extremely strong software engineering skills. Proficiency in Python and related ML frameworks such as Tensorflow, TF-Serving, JAX, and XLA/MLIR. Experience writing kernels for GPUs using CUDA. Experience using large-scale distributed training strategies. Familiarity with autoregressive sequence models, such as Transformers. ...

reviewing designs and implementing features. A crucial aspect of your role will be building and nurturing strategic relationships with open-source communities, particularly with JAX, OpenXLA, and PyTorch/XLA. You'll work closely with these communities to align framework development roadmaps with Neuron's strategic objectives, ensuring ...

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

About Graphcore Graphcore is one of the world’s leading innovators in Artificial Intelligence compute. It is developing hardware, software and systems infrastructure that will unlock the next generation of AI breakthroughs and power the ...

Trainium ML accelerators. This comprehensive toolkit includes an ML compiler, runtime, and application framework that seamlessly integrates with popular ML frameworks like PyTorch and JAX enabling unparalleled ML inference and training performance. The Inference Enablement and Acceleration team is at the forefront of running a wide range of models … Experience optimizing inference performance for both latency and throughput on such large models across the stack from system level optimizations through to Pytorch or JAX is a must have. You can learn more about Neuron Key job responsibilities This role will help lead the efforts in building distributed inference ...

The era of pervasive AI has arrived. In this era, organizations will use generative AI to unlock hidden value in their data, accelerate processes, reduce costs, drive efficiency and innovation to fundamentally transform their businesses ...

Waymo is an autonomous driving technology company with the mission to be the world's most trusted driver. Since its start as the Google Self-Driving Car Project in 2009, Waymo has focused on building the Waymo Driver-The World's Most Experienced Driver -to improve access to mobility … Architecture, Deep Learning, Reinforcement Learning, High Performance Computing, Compilers, and/or Computer Architecture. Hands-on experience with deep learning frameworks (e.g., PyTorch, TensorFlow, JAX) and ML coding, proficiency in Python and/or C/C++ We Prefer: Experience with large-scale model training, including large diffusion models ...

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 … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

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 … Experience training or deploying multimodal or diffusion models Experience in large-scale data processing (Apache Arrow, Spark, Ray) Prior work in a research lab (Google Brain, DeepMind, FAIR, Anthropic, OpenAI) What You'll Get Extreme ownership and autonomy from day one - you define and build key model systems. Founding ...

finance with our AI-native research and task automation platform, backed by leading, global venture investors. Founded by a team of experts from Google DeepMind, Meta AI, and J.P. Morgan, Finster AI provides cutting-edge solutions to help finance professionals unlock unique insights with greater efficiency and accuracy. … degree in Computer Science or relevant fields A strong programming experience and familiarity with Python based deep learning frameworks like Pytorch, JAX, Tensorflow Experience writing production ready code Familiarity with recent developments and literature on Large Language Models and AI agents Knowledge of core machine learning concepts ...

AMLZ Recruiting is the executive search brand of AMLZ GmbH, headquartered in Wiesbaden, Germany. We specialize in placing exceptional professionals and executives with leading companies worldwide. Backed by a global network and deep industry expertise ...

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