11 of 11 Remote Permanent Google JAX Jobs

Science, Machine Learning, Mathematics, or a closely related field. Strong foundational knowledge of machine learning and deep learning , with hands-on experience using frameworks such as PyTorch , TensorFlow , or JAX . Exposure to LLM fine-tuning and/or experience working with transformer-based models (e.g. GPT, LLaMA, Mistral). Interest in or familiarity with agentic AI frameworks and More ❯

learn, etc.). Comfort working with large and complex datasets, including data wrangling and exploratory analysis. Experience (or strong interest) in GPU-powered model development (e.g. using PyTorch, TensorFlow, JAX, or similar). Demonstrated ability to work independently on real-world projects. Strong communicator, with the ability to share ideas, take feedback, and collaborate effectively. A master’s or More ❯

learn, etc.). Comfort working with large and complex datasets, including data wrangling and exploratory analysis. Experience (or strong interest) in GPU-powered model development (e.g. using PyTorch, TensorFlow, JAX, or similar). Demonstrated ability to work independently on real-world projects. Strong communicator, with the ability to share ideas, take feedback, and collaborate effectively. A master’s or More ❯

facing developer workflows. Experience developing for and on Arm architectures. Experience profiling or debugging ML workloads. Experience designing, training, optimizing, and deploying machine learning models using frameworks like PyTorch, JAX, LiteRT or HuggingFace Transformers. Experience collaborating closely with product or design teams. In Return: Our team offers an inclusive and highly collaborative environment, with work that spans a wide More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … or engineering roles. Strong expertise in Machine Learning and Deep Learning, with exposure to Reinforcement Learning as a plus. Proficiency in Python and modern ML libraries (e.g., TensorFlow, PyTorch, JAX, or Keras). Experience with version control systems (GitHub, GitLab) and knowledge of clean, maintainable coding practices. Familiarity with CI/CD pipelines for automating ML workflows. Ability to More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … language processing and computer vision) in industry. Master, PhD degree or equivalent experience in applied mathematics, computer science, bioinformatics or related scientific fields. Proficiency in software engineering (Python, Pytorch, JAX, Docker, Linux). A willingness to learn and develop skills in computational genomics including building an understanding of biological concepts related to cancer. Excellent communication skills in English. Appropriate More ❯

NeurIPS, ICML, ICLR, ACL, EMNLP, 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, ) Experience in ML Research beyond completing a PhD (e.g. supervision, industry experience, leading academic initiatives, ). Excellent communication skills to report and present research findings and developments clearly … code contributions to popular open-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 and technologies. Plenty of data More ❯

e.g., NeurIPS, ICML, ICLR, ACL,EMNLP, NAACL, ICLR) with specific focus on agent systems, tool use, or multi-agentcoordination Familiarity with one or more deep learning frameworks (e.g. pytorch, jax, tensorflow, ) Excellent communication skills to report and present research findings anddevelopments clearly, both orally and in writing. Curious and innovative disposition capable of devising novel, well-founded algorithmicsolutions to … popular open-source libraries or writing production code. Experience training large-scale models over distributed nodes with cloud tools and providers such as Amazon AWS, MS 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 and technologies. Plenty of data More ❯

The world will be unrecognisable in 5 years. Machine learning models are driving our cars , testing our eyesight , detecting our cancer , giving sight to the blind , giving speech to the mute , and dictating what we consume, enjoy, and think . More ❯

at the forefront of innovation. With strategic offices in major cities worldwide, including London, Paris, Berlin, Tunis, Kigali, Cape Town, Boston, and San Francisco, InstaDeep collaborates with giants like Google DeepMind and prestigious educational institutions like MIT, Stanford, Oxford, UCL, and Imperial College London. We are a Google Cloud Partner and a select NVIDIA Elite Service Delivery Partner. … tools, and hardware optimisation techniques Experience with distributed training frameworks (Ray, Dask, PyTorch Lightning, etc.) Expertise with Python and/or C/C++ Development with machine learning frameworks (JAX, Tensorflow, PyTorch etc.) Passion for profiling, identifying bottlenecks, and delivering efficient solutions. Highly Desirable Track record of successfully scaling ML models. Experience writing custom CUDA kernels or XLA operations. More ❯

Life at UiPath The people at UiPath believe in the transformative power of automation to change how the world works. We're committed to creating category-leading enterprise software that unleashes that power. To make that happen, we need people More ❯