101 to 125 of 147 Google JAX Jobs

running experiments, and debugging pipelines, and who can carry research results through to production with engineering. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow). Ability to communicate clearly, collaborate across teams, and align research work with product and engineering priorities. Nice to have Experience quantifying ...

experience with other robotics simulators (Drake, MuJoCo, Newton, Isaac, RaiSim, Jiminy, Brax). PyTorch, our user-facing interface. Other JIT compilers such as Triton, JAX, or Numba. 3D tooling and interactive editors (Maya, Blender, Houdini). Serialization formats (USD, glTF, MJCF/URDF). C++, CUDA, or Metal. Developer ...

research direction with strong execution, and experience mentoring others on a fast‐moving, applied research team. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Nice to have Demonstrated experience fine‐tuning ...

silicon. What they're looking for: Degree in Physics/Engineering/Electronics and a PhD in Machine Learning Strong Python + PyTorch/JAX skills Publication record in ML or EDA venues Ability to move between research and implementation Clear communication across ML and circuit‐design audiences Experience ...

silicon. What they're looking for: First degree in relevant STEM subject 3+ years in industry building production ML systems Python + PyTorch/JAX expertise for training, inference, and model export Experience building data pipelines for complex scientific/engineering datasets Solid software engineering fundamentals - testing, CI/ ...

skills. Ability to balance strategic thinking with hands-on technical leadership. Comfortable operating within a fast-paced, high-growth environment. Technology Environment Python PyTorch JAX Large Language Models (LLMs) GPU Training & Inference Infrastructure Distributed ML Systems Why Join? Help shape the future of AI-powered productivity. Work alongside ...

models — architecture, training, and inference. Experience solving real problems using reinforcement learning with deep neural networks in any domain. Strong Python + PyTorch/JAX; you can profile, debug numerics, and write maintainable research code. You are self‐driven, pro‐active, communicate efficiently, document experiments clearly and communicate trade ...

lead complex research efforts while staying grounded in product impact, user experience, and real‐world performance. Strong coding and experimentation skills (Python, PyTorch/JAX, audio processing libraries). Ability to communicate clearly, collaborate across teams, and align research work with product and engineering priorities. Proven experience mentoring others ...

optimize the performance of our custom hardware. Additionally, you will drive innovations that allow the profiler to support multiple frameworks, such as PyTorch, JAX, and XLA. A successful candidate will have experience in architecting, building, and operating distributed systems with a focus on high availability and fault tolerance, Hands ...

integrating ML systems into production while staying grounded in product impact and real-world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

with strong software engineering practicesDemonstrated expertise in building production-quality code and data pipelines for ML systemsProficiency in modern AI development frameworks including: PyTorch, Jax , HuggingFace Transformers, LLM APIs (litellm etc) and vLLM for building and deploying large-scale AI applicationsUnderstanding of LLM training methodologies including instruction fine-tuning … data generation, data curation, or curriculum learning pipelinesExperience training large-scale models over distributed nodes with cloud tools such as AWS, MS Azure, or Google CloudHands-on experience with MLOps, experiment tracking, and model deployment systemsStrong interest in staying current with ML research literature and ability to quickly implement ...

C++ or Java A strong background in the mathematical foundations of core ML algorithms, paired with hands‐on experience using deep learning frameworks (PyTorch, JAX, TensorFlow), agentic orchestration ecosystems (LangGraph, Google ADK), and Model Context Protocol tooling (RustMCP) Strong written and verbal communication skills to document your work ...

integrating ML systems into production while staying grounded in product impact and real‐world quality. Strong coding and experimentation skills (Python, PyTorch/JAX/Tensorflow), and the ability to communicate clearly and align research with product and engineering priorities. Ability to lead complex research efforts, to communicate clearly ...

design and commitment to the development of reusable, scalable, and testable software components. Basic understanding of at least one major deep learning framework (PyTorch, JAX, TensorFlow). Knowledge of command‐line tools and shell scripting. Knowledge of software engineering best practices, including continuous integration (CI) and continuous deployment … issues in live systems. Experience designing and administering SQL databases, schema design, query performance, and day‐to‐day operational management. Hands‐on experience with Google Cloud Platform, in particular the services we build on: Cloud Run, Google Kubernetes Engine, Cloud Storage, Artifact Registry, Cloud SQL. Fluency in English. ...

performance, with no prior finance requirement. Candidates hold a Master’s or PhD in ML, CS, or statistics, and excel in Python, Torch/JAX/TF, #J-18808-Ljbffr ...

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

NAACL, ICLR) with specific focus on agent systems, tool use, or multi-agent coordination. Familiarity with one or more deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ...). Excellent communication skills to report and present research findings and developments clearly, both orally and in writing. Curious and innovative disposition … writing production code. Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, Microsoft Azure, LambdaLabs or Google Cloud. You will enjoy Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods ...

EMNLP, NAACL, ICLR) ) with specific focus on training data curation, synthetic data generation, etc.* Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ...)* Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ...).* Excellent communication skills to report … source libraries or writing production code.* Experience training large-scale models over distributed nodes with cloud tools such as Amazon AWS, MS Azure, or Google Cloud.*You will enjoy:** Learning and development: On-the-job coaching and learning as well as the opportunity to work with cutting-edge methods ...

this role, you will be responsible for building next generation Neuron compiler which transforms ML models written in ML frameworks (e.g, PyTorch, TensorFlow, and JAX) to be deployed AWS Inferentia and Trainium based servers in the Amazon cloud. You will be responsible for solving hard compiler optimization problems …/Vector engines/ML-accelerators. - Experience with OpenSource compiler toolset like LLVM/MLIR. - Experience with the following technologies: PyTorch, OpenXLA, StableHLO, JAX, TVM, deep learning models, and algorithms. - Experience with modern build systems like Bazel/CMake. Amazon is an equal opportunity employer and does not discriminate ...

this role, you will be responsible for building next generation Neuron compiler which transforms ML models written in ML frameworks (e.g, PyTorch, TensorFlow, and JAX) to be deployed AWS Inferentia and Trainium based servers in the Amazon cloud. You will be responsible for solving hard compiler optimization problems … degree in computer science or equivalent - Proficiency with resource management, scheduling, code generation, and compute graph optimization - Experience optimizing Tensorflow, PyTorch or JAX deep learning models - Experience with multiple toolchains and Instruction Set Architectures Amazon is an equal opportunity employer and does not discriminate on the basis of protected ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat’spossible in asset management. Today, AI is reshaping every aspect of our industry, from ...

hackajob is partnering directly with BlackRock to hire for this role. About this role AI Labs Overview For more than three decades, BlackRock has combined world-class talent and technology to redefinewhat'spossible in asset ...

Olix Computing seeks an Architect/Staff Systems Software Engineer to own the DX-1 runtime and serving stack, connecting PyTorch/JAX to the hardware and enabling production inference at rack scale. This is a whole-stack role at the intersection of runtime, network, and accelerator with close ...