101 to 125 of 225 JAX Jobs

especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). Proficiency in Python and strong experience with AI/ML frameworks (e.g., TensorFlow, PyTorch, JAX, Hugging Face Transformers). Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). Solid More ❯

Expertise in machine learning applied to proteins, small molecules, and molecular simulations. Strong programming skills in Python and proficiency with ML frameworks (e.g., PyTorch, JAX, TensorFlow). Hands-on experience with scientific computing libraries (e.g., RDKit, OpenMM) and molecular representations (e.g., SMILES, molecular graphs, 3D structures). Ability to bridge More ❯

models at scale, ideally within product environments or a data-rich business. Tech-savvy : Proficient in Python and modern ML frameworks (e.g., PyTorch, TensorFlow, JAX). Strong command of cloud- AI tooling (e.g., AWS SageMaker, GCP Vertex AI, AzureML/AzureAI, MLflow, Weights & Biases). Well-rounded Engineer : Comfortable working More ❯

. Knowledge of linear algebra, calculus and statistics equivalent to at least first-year university coursework. Experience exploring, analysing and visualising data. Experience using Jax, PyTorch, NumPy, Pandas or similar ML/scientific libraries. In addition, the following would be an advantage: Experience with advanced recent LLM techniques. Scientific domain More ❯

and algorithm scaling, or neural network architectures. Experience communicating complex research for public audiences of peers. Experience with deep learning frameworks such as Pytorch, Jax, or Tensorflow. Must obtain work authorization in country of employment at the time of hire, and maintain ongoing work authorization during employment. Preferred Qualifications Proven More ❯

models at scale, ideally within product environments or a data-rich business. Tech-savvy : Proficient in Python and modern ML frameworks (e.g., PyTorch, TensorFlow, JAX). Strong command of cloud-native AI tooling (e.g., AWS SageMaker, GCP Vertex AI, AzureML/AzureAI, MLflow, Weights & Biases). Well-rounded Engineer : Comfortable More ❯

with evaluating and improving the generalisability and robustness of ML systems. Proficiency in programming languages such as Python and ML frameworks (e.g., PyTorch, TensorFlow, JAX). Excellent communication skills to collaborate effectively with cross-functional teams and present findings. One or more papers at top-tier venues (such as NeurIPS More ❯

if you: Have 5+ years of industry-related experience Are proficient in Python and have experience with deep learning frameworks such as PyTorch or Jax Have a strong software engineering background and are interested in working closely with researchers and other engineers Enjoy pair programming (we love to pair!) Care More ❯

machine learning infrastructure, including technologies like MLFlow, Kubeflow and large scale distributed systems - Expertise in deep learning frameworks and tensor libraries like PyTorch, Tensorflow, JAX/FLAX Suggested Skills: -ML Algorithm Development -Machine Learning and Deep Learning -Information retrieval/recommendation systems -Technical leadership LinkedIn is committed to fair and More ❯

efficiency, accuracy, and hardware utilization. Implement distributed training strategies at scale. Apply advanced techniques like sparsity, quantization, and distillation. Use frameworks like TensorFlow, PyTorch, JAX, and Hugging Face Transformers. Benchmark and profile models on custom accelerators. Stay current with AI and hardware advancements and contribute to ongoing R&D efforts. More ❯

especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). Proficiency in Python and strong experience with AI/ML frameworks (e.g., TensorFlow, PyTorch, JAX, Hugging Face Transformers). Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). Solid More ❯

especially Vertex AI (Workbench, Training, Prediction, Pipelines, Feature Store). Proficiency in Python and strong experience with AI/ML frameworks (e.g., TensorFlow, PyTorch, JAX, Hugging Face Transformers). Experience with data engineering and building robust data pipelines for AI/ML workloads on GCP (e.g., BigQuery, Dataflow). Solid More ❯

both a scientist and an engineer: excited by innovation, and pragmatic about delivering impact! 🛠️ Key Technologies: Python and modern ML libraries (e.g. PyTorch, TensorFlow, JAX) AWS and cloud infrastructure for scalable training and inference ML ops tooling, model monitoring, and data pipelines Bonus: experience with LLMs & generative AI is a More ❯

both a scientist and an engineer: excited by innovation, and pragmatic about delivering impact! 🛠️ Key Technologies: Python and modern ML libraries (e.g. PyTorch, TensorFlow, JAX) AWS and cloud infrastructure for scalable training and inference ML ops tooling, model monitoring, and data pipelines Bonus: experience with LLMs & generative AI is a More ❯

quality ML systems. Preferred Qualifications: Master's degree or PhD in Computer Science or a related technical field. Experience with modern ML frameworks (e.g., JAX, Pytorch or TensorFlow). Experience with launching applied Machine Learning/Natural Language Processing (ML/NLP) projects. About the Job Google's software engineers More ❯

SSMs) Solid development skills in Python and/or C++ Familiarity with ML libraries/frameworks such as PyTorch (preferred), TensorFlow, and/or JAX Intellectual curiosity, versatility, and originality combined with a pragmatic outlook Ability to reason through quantitative problems and communicate effectively with trading researchers Reliable and predictable More ❯

with evaluating and improving the generalisability and robustness of ML systems. Proficiency in programming languages such as Python and ML frameworks (e.g., PyTorch, TensorFlow, JAX). Excellent communication skills to collaborate effectively with cross-functional teams and present findings. One or more papers at top-tier venues (such as NeurIPS More ❯

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU, NPU More ❯

experience in academia or industry We will consider various levels of experience in relevant research areas Key Skills: Experience with ML frameworks (PyTorch, TensorFlow, JAX) and efficient ML (incl. quantization, pruning, sparsification, distillation, etc.) Experience with deployment on embedded/mobile devices (such as smartphones, with mobile CPU, GPU, NPU More ❯

including designing new architectures, hands-on experimentation, analysis, and visualisation. Strong knowledge of linear algebra, calculus and statistics. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas. A passion for applying ML research to real world problems. Depending on your More ❯

experience with prompt engineering and post-training techniques for large language models, including data pipeline and evaluation methodologies. Experience with ML frameworks such as JAX, PyTorch, or TensorFlow. Experience training and deploying large-scale models At Google DeepMind, we value diversity of experience, knowledge, backgrounds and perspectives and harness these More ❯

reinforcement learning, non-convex optimisation, approximate inference, NLP, or Bayesian methods Strong programming skills, ideally in Python, with experience using tools like NumPy, Pandas, JAX, PyTorch or TensorFlow A strong publication record in top-tier venues (e.g., NeurIPS, ICML, ICLR) or competitive performance in ML challenges such as Kaggle or More ❯

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 More ❯

industry, including knowledge of the landscape and Pure’s solutions with a focus on the following: Deep understanding of industry AI frameworks (pytorch, tensorflow, jax etc) Deep understanding on high performance, low latency, highly parallelized compute solutions at scale (compute, network, storage) Deep understanding of accelerated compute environments (GPU, ASIC More ❯

including designing new architectures, hands-on experimentation, analysis, and visualisation. Strong knowledge of linear algebra, calculus and statistics. Experience using ML frameworks such as JAX, PyTorch, or TensorFlow, and scientific software such as NumPy, SciPy, or Pandas. A passion for applying ML research to real world problems. Depending on your More ❯