11 of 11 Google JAX Jobs

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

experience. Strong expertise in deep learning, neural networks, and generative models (diffusion, GANs, transformers). Hands-on experience with modern ML frameworks (e.g., PyTorch, JAX, TensorFlow). Advanced programming skills in Python. Solid understanding of model training, evaluation, and scaling. Strong problem-solving, analytical, and communication skills. Ability ...

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

machine learning, statistics and/or mathematics.Strong programming skills in Python. Ideally, also proficiency in at least one major deep learning framework (e.g., PyTorch, JAX).An early track record of research (e.g., a high-quality Master's thesis, open-source contributions, or workshop papers). Excellent command of written ...

ICLR. Hands-on experience applying deep learning models to complex, ideally temporal or financial, datasets. Expertise in Python and modern ML frameworks (TensorFlow, PyTorch, JAX, etc.). Prior experience in a commercial or industry setting post-PhD, ideally in a similarly data-driven domain. Why Apply Compensation and total ...

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

Skills Experience with rendering at scale (e.g. render farms) Knowledge of computer vision, particularly real-time systems Experience with machine learning frameworks (e.g. PyTorch, JAX) Familiarity with the Blender C++ codebase Experience with AI-assisted coding or agent-based workflows Strong visual judgment and attention to detail for high ...

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

generative models — architecture, training, and inference. Experience with deep learning infrastructure: streaming datasets, checkpointing & state management, distributed training strategies. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You document experiments clearly and communicate trade‐offs crisply. Nice-to-Have: Robotics ...

Silicon Valley office | Reports to Founders Introduction: Join us at Fastino as we build the next generation of LLMs. Our team, boasting alumni from Google Research, Apple, Stanford, and Cambridge is on a mission to develop specialized, efficient AI. Fastino's GLiNER family of open source models has been … comparable generative AI architectures Optional - Comprehensive technical proficiency and practical experience with leading deep learning frameworks, including advanced competency in one of PyTorch, JAX, TensorFlow, or equivalent platforms for model development and optimization ...

frontier AI company backed by NVIDIA, building large-scale open-weight foundation models alongside researchers and engineers from DeepMind, OpenAI, Meta, Anthropic, and Google Brain. ⚡ What You’ll Do Optimise GPU performance and training efficiency across 1,000+ GPU clusters Improve utilisation, throughput, and reliability across distributed training infrastructure … accelerate large-scale model training 🔧 What They’re Looking For Deep GPU infrastructure/distributed systems experience Strong knowledge of CUDA, NCCL, PyTorch, DeepSpeed, JAX, Megatron-LM, vLLM, etc. Experience operating large-scale GPU clusters (1,000+ GPUs) Kubernetes, Slurm, or similar orchestration expertise BONUS: Experience working on NVIDIA ...