Latest TensorFlow Jobs

Senior Machine Learning Applications and Compiler Engineer, LPX

Develop high-performance compiler and runtime components for NVIDIA's LPX inference stack, optimizing neural network workloads on future spatial processors. Collaborate with hardware teams to co-design features and implement end-to-end optimizations across compiler, runtime, and deployment layers. Focus on graph transformations, scheduling, and memory layout for domain-specific AI hardware.

NVIDIA

Hybrid Permanent

Senior HPC Performance Engineer

Conduct performance characterization and analysis on large multi-GPU and multi-node clusters, focusing on GPU communication libraries like NCCL and NVSHMEM. Evaluate system-level interactions across hardware and software stacks, develop tools to visualize performance data, and debug issues across the full stack. Work closely with cross-functional teams to optimize communication performance at scale for HPC and deep learning applications.

NVIDIA PLN 221,250 – PLN 507,000 pa

Remote Permanent

Research Scientist (Machine Learning), Lausanne

A Research Scientist in machine learning will develop novel AI models and algorithms to accelerate drug discovery, working in interdisciplinary teams to solve complex problems in computational biology and chemistry. The role involves designing deep learning architectures, training and scaling models, and translating research into practical applications. Collaboration with scientists across disciplines is central to advancing Isomorphic Labs’ mission of solving disease through AI.

Isomorphic Labs United Kingdom

Hybrid Permanent

Senior Security Engineer (AI Safety), London, Lausanne

This role involves securing AI-driven drug discovery systems by designing risk frameworks, protecting machine learning artifacts, and implementing security controls across AI infrastructure. The engineer will lead incident response for AI-specific threats, enforce compliance with global regulations, and develop safety protocols for autonomous agentic workflows and large language models within a high-stakes scientific environment.

Isomorphic Labs United Kingdom

Hybrid Permanent

ML Research Engineer, London

Develop and optimise cutting-edge AI models at the intersection of machine learning and drug discovery, translating research into scalable, production-ready systems. Work closely with scientists to build predictive and generative models using advanced deep learning techniques, with a focus on innovation in computational biology and chemistry. Contribute to a collaborative, interdisciplinary team advancing a digital biology platform for transformative medicine development.

Isomorphic Labs London, United Kingdom

On-site Permanent