1 to 25 of 28 Google JAX Jobs

Senior AI Engineer - Google AI & Generative Intelligence Job Number: 26-00855 Ready for a rewarding opportunity in the Financial Services Industry? ECLARO is looking for a Senior AI Engineer - Google AI & Generative Intelligence for our client in Paramus, NJ. ECLARO's client is a market-leading insurance company … challenge, then take a chance at this rewarding opportunity! Position Overview: A highly experienced Senior AI Engineer with deep expertise in Google AI technologies, Generative AI. Brings 10-15 years of broad software engineering experience, with the last 4 years focused exclusively on Artificial Generative Intelligence, including designing, building ...

optimization with experience in programming languages such as Python (primary), Java and SQL Familiarity with deep learning tools and frameworks such as TensorFlow, Pytorch, Jax, Hugging libraries etc. Expert knowledge with traditional (tabular) ML modeling and methods. Proven proficiency in deep learning methodologies, fine tuning, and transformer architectures. ...

optimization with experience in programming languages such as Python (primary), Java and SQL Familiarity with deep learning tools and frameworks such as TensorFlow, Pytorch, Jax, Hugging libraries etc. Expert knowledge with traditional (tabular) ML modeling and methods. Proven proficiency in deep learning methodologies, fine tuning, and transformer architectures. ...

optimization with experience in programming languages such as Python (primary), Java and SQL Familiarity with deep learning tools and frameworks such as TensorFlow, Pytorch, Jax, Hugging libraries etc. Expert knowledge with traditional (tabular) ML modeling and methods. Proven proficiency in deep learning methodologies, fine tuning, and transformer architectures. ...

Physics, Applied Mathematics, Statistics, Computer Science or similar Excellent Python skills for AI/ML development Hands-on experience with frameworks such as PyTorch, JAX, TensorFlow, Scikit-Learn or similar Strong mathematical and statistical foundations Experience developing ML models or AI systems beyond experimentation An ability to work across ...

language processing, including hands-on experience with transformer architectures (e.g., GPT, BERT, T5). Expertise in deep learning frameworks such as TensorFlow , PyTorch , or JAX . Strong knowledge of machine learning algorithms and their application to NLP tasks. Experience with cloud platforms such as AWS, GCP, or Azure ...

learning, and/or large language models (LLMs) Experience in chip design and applied AI research Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience influencing organizational policy and securing major research funding ...

learning, and/or large language models (LLMs) Experience in chip design and applied AI research Familiarity with deep learning frameworks (e.g., TensorFlow, PyTorch, JAX) Experience influencing organizational policy and securing major research funding ...

probability, optimisation, statistics, and ML fundamentals • Strong Python programming skills, with C++ experience considered highly beneficial • Experience using frameworks such as PyTorch, TensorFlow, or JAX • Experience building scalable ML systems for production environments • Familiarity with cloud or GPU based training infrastructure and MLOps workflows • Experience designing evaluation, testing ...

awareness of latency and cost tradeoffs MLOps competency — experiment tracking, model versioning, evaluation pipelines, ML CI Experience with Python and modern ML frameworks (PyTorch, JAX) Excellent verbal and written English communication skills Excellent analytical and problem-solving skills A desire to innovate and push current practice Desirable: Experience with ...

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 If this sounds like ...

abilities, including technical documentation 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 If this sounds like ...

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

boundaries of what AI can actually do. Experience required: Strong hands-on experience with Reinforcement Learning (RL) Python and modern ML frameworks ( PyTorch, JAX, TensorFlow ) Experience working with time-series or sensor data Ability to turn real-world problems into practical ML solutions Comfortable taking models from research into ...

research projects end to end — from problem formulation through to evaluation and deployment. Knowledge of modern, larger-scale Python and the ML stack (PyTorch, JAX, or equivalent). You write research-grade code. Practical experience building large-scale data pipelines. We don't have data infrastructure — you'll help ...

About the Role JD.com seeking an experienced AI Engineer to join our growing technical team at Joybuy (https://www.joybuy.com/). Joybuy is JD.com’s European full-category online retail brand designed to ...

reinforcement learning Experience with agent frameworks, tool-use planning and workflow orchestration Systems-level fluency across data pipelines, training infrastructure and inference (PyTorch, JAX, TensorFlow) Strong MLOps discipline: CI/CD for models, offline/online evaluation, safety and guardrails The ability to influence at exec level and mentor ...

message types, and topic debugging* Experience working with physical robot hardware, camera streams, joint states, or latency-sensitive control paths* OpenCV, NumPy, pandas, PyTorch, JAX, cloud, HPC, or LeRobot dataset format experience* Experience working closely with ML researchers or robotics engineers in a fast-moving technical environment The Company ...

knowledge of ML evaluation, validation strategy, and failure modes including error analysis and iteration planning ● Experience with model development using PyTorch/TensorFlow/JAX ● Ability to communicate clearly across technical + scientific stakeholders ● Strong experience leading or coordinating cross-functional teams, and thinking strategically across product and science ...

work in bwHPC-S5. Ideally, your profile is supplemented by: Experience with AI frameworks, workloads (e.g. Natural Language Processing), and libraries (e.g., PyTorch or JAX). Experience with parallel programming paradigms. Some background in the field of (green) transport (e.g., fleet optimisation, energy-efficient logistics, smart city planning). ...

learning systems, with shipped models or published research. Hands-on experience with Vision Language Models, LLMs, or generative models. Strong Python and PyTorch/JAX skills, including debugging, profiling, and shipping. A passion for bridging the gap between simulation and reality. This is a technically challenging role with ...

agents already running in production What We're Looking For Essential 3-5 years training and deploying deep RL agents in Python PyTorch or JAX, and RL libraries such as Gymnasium A background in physical systems - engineering (mechanical, electrical, structural, biomedical), physics, robotics, autonomous driving, or control systems ...

3. A working knowledge and practical experience with Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL. Key Words: Machine Learning Engineer/Data Processing/Model Building/NMR/Deep Learning/Computational Biophysics/ ...

deep learning and modern foundation model architectures Experience with probabilistic modelling techniques such as Bayesian methods, Gaussian Processes, or variational inference Expertise in PyTorch, JAX, or similar frameworks Experience training large models on distributed infrastructure Strong software engineering and systems design capabilities Nice to Have Background in computational biology ...

expertise in foundation models, transformers, representation learning, and generative AI Proven experience training large-scale models on distributed GPU infrastructure Deep knowledge of PyTorch, JAX, or equivalent deep learning frameworks Strong software engineering and systems design capabilities Experience leading complex technical initiatives and mentoring engineering teams Expertise in probabilistic ...