6 of 6 Google JAX Jobs

others); OR equivalent work experience delivering state-of-the-art AI/ML solutions. Experience with at least one major deep learning framework (PyTorch, JAX, TensorFlow). Familiarity with machine learning literature and state-of-the-art approaches. Experience developing and delivering robust software solutions, including demonstrated advanced programming ...

applied mathematicsExperience training, fine-tuning or deploying production-grade modelsExcellent programming skills in Python, with Rust, C++ or Java advantageousStrong experience with PyTorch, JAX or TensorFlowKnowledge of LLMs, transformer architectures, model optimisation or agentic systemsThe ability to turn complex research into reliable, production-ready softwareBonus points ...

basic deployment, ideally demonstrated by published software projects (e.g., github).Experienced in developing and implementation of deep learning systems (e.g., in PyTorch or JAX) and agentic software design and testing.Experienced in modelling or simulation of molecules, ideally biomoleculesHigh-quality publications in leading disciplinary or interdisciplinary venues or demonstrated leadership ...

easy or obvious solutions. QualificationsStrong engineering skills, especially any of: CUDA/Triton/Pallas/CuTe DSL kernel development, lower-level PyTorch/JAX/XLA development, CUDA Graphs, FPGA/ASIC experienceMust have two or more years work experience building deep learning systems, for any domain: robotics ...

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

Full TimeOn-siteCambridge, UKPosted 16 hours ago60,000-120,000 GBP/Year Machine Learning Engineer job in Cambridge | Salary: 60,000-120,000Hexwired has partnered with an innovative technology company developing advanced AI and ...