AI Performance Engineer (Life Sciences)
HelsinkiCompetitive0 applicants
About this role
ADVANCE YOUR CAREER. ADVANCE THE WORLD.
At AMD, we believe technology has the power to solve the world's most important challenges. From advancing healthcare and scientific discovery to powering AI and the technologies people rely on every day, innovation at AMD is shaping the future.
Whether you're designing next-gen processors, enabling AI breakthroughs, or bringing leading edge products to market, every role at AMD contributes to something bigger - technology that moves the world forward. Join us and, together, we'll advance your career.
Responsibilities
- We are looking for an AI Performance Engineer, Life Sciences to contribute to the development and optimization of foundation models and agentic AI systems for drug discovery. You will work within a multidisciplinary team of machine learning researchers, computational chemists, and software engineers to deliver AI systems with real-world therapeutic impact. This role offers the opportunity to apply expertise in deep learning performance optimization, collaborate across technical domains, and deliver production-grade software solutions.
- MAIN RESPONSIBILITIESOptimize machine learning models through profiling, bottleneck analysis, and efficient mapping of workloads to GPU architectures.
- Analyze emerging machine learning models, understand their compute and memory requirements, and optimize them for training and inference across a range of hardware platforms.
- Perform optimization at the kernel, framework, and hardware levels.
- Work hands-on with deep learning and transformer architectures, as well as physics-based simulation models.
- Profile and analyze workloads on current hardware, identify performance bottlenecks, and develop strategies to improve efficiency and scalability.
- COLLABORATIONContribute to the Life Sciences workstream within AMD Silo AI's R&D and Models unit as part of an interdisciplinary team.
- Partner with life sciences AI developers and domain experts to understand requirements and translate them into