1 to 25 of 33 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 ...

Strong software engineering skills: clean code, testing, debugging, version control, and system design. Proficiency in Python with experience in ML frameworks (PyTorch, TensorFlow, or JAX). Hands‐on experience with cloud platforms (AWS, GCP, or Azure) and containerisation (Docker, Kubernetes). Solid understanding of ML fundamentals (model training, evaluation ...

/ML solutions in industry or engineering environments. Strong expertise in Python and AI/ML libraries: PyTorch, TensorFlow, Scikit‐Learn, Keras, NumPy, Pandas, JAX, XGBoost, PyTorch Geometric, OpenCV, Transformers, LangChain, Pinecone . Proven hands‐on experience deploying models in production and building scalable AI software. Deep understanding ...

equivalent experience in Computer Science, Maths, Machine Learning, Data Science, or related field. Desirable: Exposure to machine learning frameworks such as PyTorch, JAX, Triton, TensorFlow Experience with distributed workload management systems such as Kubernetes, VLLM, Keras or MLOps pipelines Experience working with hardware simulators or emulators (e.g. QEMU). ...

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

/ML incubations or complex deep‐tech cross‐functional initiatives with measurable outcomes. Expertise in Python and deep experience with deep‐learning frameworks (e.g., Jax, TensorFlow, PyTorch) and state‐of‐the‐art ML techniques. Practical experience with modern AI paradigms such as world models, foundation models, multi‐modal language ...

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

Computer Science (or equivalent proven track record). 4 + years working on large‐scale ML codebases. Hands‐on with PyTorch, JAX or TensorFlow; comfortable with distributed training (DeepSpeed/FSDP/SLURM/K8s). Experience in deep learning, NLP or LLMs; bonus for CUDA or data‐pipeline ...

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

have been instrumental in: building machine learning models and pipelines in Python, using common libraries and frameworks (PyTorch/CUDA, ideally with exposure to JAX, NumPy/SciPy), especially including deep learning applications; developing models for bespoke problem settings that involve high‐dimensional data (spatiotemporal, geometric, physical); iterating ...

machine learning models for time-series or sensor data Strong programming skills in Python Experience with machine learning frameworks such as PyTorch, TensorFlow, or JAX Experience building data pipelines and training workflows Ability to work with imperfect, noisy, or incomplete datasets Nice to Have Experience with physiological signal processing ...

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 Track record of contributing ...

members in diverse fields Strong programming abilities in Python, and extensive experience with the scientific and machine-learning stack: Numpy, Torch/Tensorflow/Jax, Scikit-learn, Polars, SQL Expertise with deep learning approaches such as diffusion or flow matching Proficiency in modern software development practices: code testing, documentation ...

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

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

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

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

Hands‐on experience training, fine‐tuning, or deploying large‐scale ML models in production. Proficiency with at least one modern ML framework (e.g. PyTorch, JAX), and ability to learn others quickly. Experience with distributed training and inference frameworks (e.g. DeepSpeed, FSDP, Megatron, ZeRO, Ray). Strong software engineering fundamentals ...

excited by applying those skills to one of the most interesting data domains in AI. Areas of interest include: PyTorch/JAX internals CUDA, XLA or Triton optimisation Distributed training systems GPU cluster performance tuning Large-scale experiment management Training reliability and reproducibility Kubernetes and orchestration tooling Infrastructure supporting ...

work with minimal friction. Iterations on models and systems are measurable, safe, and improve user experience over time. Tech Stack Python PyTorch/JAX GPU-based training and inference system Ideal Experience You have built or shipped real ML systems used by people, not just demos. You are comfortable ...

software engineering practices* Demonstrated expertise in building production-quality code and data pipelines for ML systems* Proficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applications* Understanding of LLM training methodologies including instruction fine … generation, data curation, or curriculum learning pipelines* Experience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google Cloud* Hands-on experience with MLOps, experiment tracking, and model deployment systems* Strong interest in staying current with ML research literature and ability ...