1 to 25 of 38 Remote/Hybrid Google JAX Jobs

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic ...

related field, with at least 8 years of experience. Strong programming skills in Python and proficiency with Deep Learning frameworks (e.g., PyTorch, JAX, TensorFlow). Demonstrated hands-on experience implementing and deploying deep learning models in production or research environments. Strong scientific rigor, curiosity, and ability to perform bibliographic ...

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

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

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...

Role: AI Engineer Experience: 8-10years Location: London Contract duration: 12 months Job Description: Proven experience as an AI, Machine Learning, or Privacy Engineer. • Proficiency in programming languages such as Python is added advantage. ...