26 to 47 of 47 Permanent Google JAX Jobs

collaborating across functions, especially with engineering, hardware, and system design teams Solid programming and prototyping skills using modern deep learning frameworks (e.g. PyTorch, TensorFlow, JAX) Location: London (hybrid, with 4 days onsite and flexible hours) This is a high-impact role for someone who thrives on building ambitious technology More ❯

in manufacturing Expertise in Deep Learning , Computer Vision , Generative AI , or Reinforcement Learning Strong programming in Python and C++ ; experience with PyTorch , TensorFlow , or JAX Experience with AI system design and data infrastructure in complex environments Fluent in French & English A proactive, solutions-oriented mindset and ability to thrive More ❯

e.g., generative models, GNNs, active learning, foundation models) in a scientific context. Strong coding skills and experience with relevant ML frameworks (e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit). Deep understanding of molecular representation techniques and their application in modeling. Experience in designing and implementing robust, scalable More ❯

e.g., generative models, GNNs, active learning, foundation models)in a scientific context.- Strong coding skills and experience with relevant ML frameworks(e.g., PyTorch, TensorFlow, JAX) and chemistry toolkits (e.g., RDKit).- Deep understanding of molecular representation techniques and their application in modeling.- Experience in designing and implementing robust, scalable More ❯

Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯

Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯

Key responsibilities: Develop cutting-edge simulation frameworks and AI infrastructure. Optimize ML models using techniques like quantization and pruning. Leverage GPU power (CUDA, Python, Jax) for accelerated training. Qualifications: Expertise in Python/C++ and deep learning (PyTorch). Experience building Deep Reinforcement Learning Models. Experience with Robotics and More ❯

You've worked with open-source models like Llama/Deepseek You know how to get into the nuts and bolts of PyTorch/JAX You have experience in applied research with a product focus Preferred but Not Required You have experience building large-scale online evaluation pipelines You More ❯

As a Machine Learning Engineer in the Applied AI team at Graphcore, you will contribute to advancing AI technology by developing and optimising AI models tailored to our specialised hardware. Working closely with the Software development and Research teams, you More ❯

Rejuvenate Biomed is a clinical-stage biotechnology company based in Diepenbeek, Belgium. Our mission is to discover and develop therapeutics to delay the onset or progression of age-related diseases. We are looking for an experienced and innovative Senior AI More ❯

Staff Software Engineer, GPU Performance, Core ML corporate_fare Google place London, UK Apply Minimum Qualifications: Bachelor's degree or equivalent practical experience. 8 years of experience in software development and with data structures/algorithms (e.g., C++ or Python). 5 years of experience with Machine Learning (ML … and performance bottlenecks. Ability to develop and utilize sophisticated performance models and benchmarks to guide optimization efforts and hardware roadmap decisions. About the Job: Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our … and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google's needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to More ❯

record in top-tier conferences (e.g., NeurIPS, ICML, ICLR, ACL, EMNLP, NAACL, ICLR). 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 … 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. Other key points: Plenty of data, compute, and high-impact problems: Our scientists and engineers get to explore large datasets and discover new More ❯

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 … by publications in reputable scientific journals or conferences. Expertise in software development and programming in Python. Experience with deep learning frameworks such as PyTorch, JAX, and/or TensorFlow. Excellent ability to communicate complex ideas clearly, both verbally and in writing, and to work effectively within a multidisciplinary team. More ❯

sales organization or public procurement organization. Experience designing and deploying with one or more from the following Machine Learning (ML) frameworks e.g., TensorFlow, PyTorch, JAX, Spark ML, etc. Experience with business use case to algorithm selection, training, tuning, inference, and training pipeline, designd and build. Knowledge of AI/… ML. About the job The Google Cloud Consulting Professional Services team guides customers through the moments that matter most in their cloud journey to help businesses thrive. We help customers transform and evolve their business through the use of Google's global network, web-scale data centers, and … the future of businesses of all sizes and use technology to connect with customers, employees, and partners. As a Cloud Solutions Architect on the Google Public Sector Rapid Innovation Team, you will partner closely, as a full stack software developer, with customers and other Google Cloud Platform (GCP More ❯

Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯

Machine Learning Researcher - LLM/VLM Are you a PhD-educated Machine Learning Researcher looking for a new opportunity? If so, our client, a global consumer electronics company, is actively expanding their team. This role is based at one of More ❯

Science, a related technical field, or equivalent practical experience. 6 years of experience with machine learning model development and deployment, frameworks (e.g., PyTorch, Tensorflow, Jax, Ray, etc.), AI accelerators (e.g., TPUs, GPUs), model architectures (e.g., encoders, decoders, transformers), and using machine learning Application Programming Interface (APIs). 6 years … National Security customers on how to configure and deploy their Artificial Intelligence (AI) and Machine Learning (ML) solutions. You will support customer implementations of Google Cloud products through architecture guidance, best practices, data migration, capacity planning, implementation, troubleshooting, monitoring, etc. You will consult with customers on how to design … to deploy solutions and deliver workshops to educate and empower customers. You will work with Product Management and Product Engineering to drive excellence in Google Cloud products and features.Google Public Sector brings the magic of Google to the mission of government and education with solutions purpose-built for More ❯

AI Research Engineer (London) A fast-growing defence tech startup focused on AI and software for autonomous systems is hiring an AI Research Engineer for their remote team. Partnered with a third-party recruiter, this well-funded company is committed More ❯

AI Research Engineer (London) A fast-growing defence tech startup focused on AI and software for autonomous systems is hiring an AI Research Engineer for their remote team. Partnered with a third-party recruiter, this well-funded company is committed More ❯

Ops lifecycle, tools, and frameworks Nice to have: Experience supporting machine learning research activities Familiarity with python-based ML frameworks like TensorFlow, PyTorch, or JAX Experience with Google Cloud Platform (GCP) Culture and values What does it take to be successful at IsoLabs? It's not about finding More ❯

Machine Learning Engineer | Research Platform: We are a tier 1 supplier for a well-known buy-side quant firm on the lookout for a Machine Learning Engineer (£475,000-£515,000) who is happy to work at the interface of More ❯

Machine Learning Engineer | Research Platform: We are a tier 1 supplier for a well-known buy-side quant firm on the lookout for a Machine Learning Engineer (£475,000-£515,000) who is happy to work at the interface of More ❯