AI Performance Engineer (Life Sciences)

CareerArc Group; Helsinki; On-site; EUR 120,000 - 180,000; Full time

Benefits offered by this job

Benefits at a glance

Job summary

AMD is seeking an AI Performance Engineer, Life Sciences to optimize foundation models and agentic AI systems for drug discovery. You will profile and optimize workloads across GPU architectures, delivering production-grade software with high efficiency.

You will collaborate with ML researchers, computational chemists, and software engineers across Finland and internationally, focusing on kernel, framework, and backend optimizations for transformer-based workloads.

Qualifications

• Master's degree or PhD, or equivalent practical experience.
• Proven experience delivering ML, simulation, or HPC solutions.
• Experience building ML and simulation applications and pipelines.
• Strong understanding of deep learning architectures including transformers.
• Proficiency in Python; production C++ experience.

Responsibilities

• Optimize ML models via profiling, bottleneck analysis, and workload mapping to GPUs.
• Analyze models, understand compute/memory needs, optimize for training and inference.
• Perform kernel, framework, and hardware level optimizations.
• Work with transformers and physics-based simulation models in production.
• Profile workloads on hardware, identify bottlenecks, improve efficiency and scalability.

Skills

Deep learning optimization

Python

C++ production

PyTorch

Transformer models

Performance profiling

AI/ML systems

English communication

Education

Master's degree or PhD in CS / related field

Tools

HIP

CUDA

Triton

TensorRT

Nsight

ROCm

Job description

ADVANCE YOUR CAREER. ADVANCE THE WORLD.

At AMD, we believetechnology 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.

Whetheryou'redesigning next-gen processors, enabling AI breakthroughs, orbringing leading edge products to market, every role at AMD contributes to something bigger technologythat moves the world forward.Join us and, together, we'll advance your career.

THE ROLE

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 RESPONSIBILITIES

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

COLLABORATION

• Contribute 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 effective technical solutions.
• Share knowledge, experience, and best practices with the wider team through training, mentoring, and collaboration.
• Collaborate with teams located in Finland and internationally across the organization.
• Engage with clients to explain performance bottlenecks, optimization opportunities, and proposed solutions clearly and effectively.

MAIN GOALS FOR THE FIRST SIX MONTHS

• Benchmark, analyze, and optimize the performance of key machine learning applications on single and multi‑GPU systems at the kernel, framework, and hardware levels.
• Design, implement, and test GPU kernels and algorithms for tensor operations, including matrix multiplication and convolutions used in high‑performance machine learning libraries and frameworks.
• Identify architectural opportunities to improve the performance of transformer‑based and deep learning architectures.
• Communicate learnings, outcomes, and impact to internal and external stakeholders in a clear, structured, and effective way.
• Deliver high‑quality, maintainable code and documentation aligned with open‑source software development best practices.

WHAT YOU'LL BRING SKILLS AND QUALIFICATIONS

• A Master's degree, PhD, or equivalent practical experience in Computer Science, Computer Engineering, Computational Science, Applied Mathematics, Cheminformatics, Bioinformatics, or a related field.
• Relevant experience developing and delivering machine learning, simulation, or high‑performance computing solutions.
• Experience building machine learning and simulation applications and pipelines.
• A strong understanding of deep learning architectures, including transformers, diffusion models, and language models.
• Proficiency in Python and familiarity with C++ in production environments.
• Experience with PyTorch; familiarity with JAX would be beneficial.
• Software engineering skills across rapid prototyping, debugging, profiling, optimization, and the delivery of maintainable production code.
• Familiarity with AI‑assisted development tools such as Claude Code or Cursor.
• Experience with performance profiling tools such as rocprof, PyTorch Profiler, or Nsight.
• Experience with framework‑level optimization, including compilation techniques and backends such as Inductor or TensorRT.
• Experience profiling and optimizing GPU workloads using ROCm and/or CUDA.
• GPU software development experience using technologies such as HIP, CUDA, or OpenCL.
• Experience developing GPU kernels with HIP, CUDA, or Triton.
• Experience working in high‑performance computing environments.
• Effective written and verbal communication and presentation skills in English.

ADDITIONAL VALUED EXPERIENCE

• Experience building and operating large‑scale machine learning systems, including training infrastructure and distributed computing environments.
• Understanding of GPU architectures and low‑level optimization techniques, including memory hierarchy, instruction scheduling, and performance trade‑offs.
• Experience developing, maintaining, and supporting open‑source software projects, including releases, documentation, and continuous integration pipelines.
• Experience with cloud platforms such as AWS, GCP, or Azure.
• Familiarity with cluster orchestration technologies such as Slurm, Kubernetes, or Yarn.
• Hands‑on experience with life sciences foundation models, including OpenFold, AlphaFold, Boltz, ESM, Geneformer, or scGPT families.
• Hands‑on experience with life sciences simulation workloads such as GROMACS or NAMD.
• A publication record at conferences such as NeurIPS, ICML, ICLR, MLSB, ISMB, or RECOMB.
• Experience building production‑grade agentic AI systems and workflows.

AMD benefits at a glance.

AMD does not accept unsolicited resumes from headhunters, recruitment agencies, or fee‑based recruitment services. AMD and its subsidiaries are equal opportunity, inclusive employers and will consider all applicants without regard to age, ancestry, color, marital status, medical condition, mental or physical disability, national origin, race, religion, political and/or third‑party affiliation, sex, pregnancy, sexual orientation, gender identity, military or veteran status, or any other characteristic protected by law. We encourage applications from all qualified candidates and will accommodate applicants' needs under the respective laws throughout all stages of the recruitment and selection process.

AMD may use Artificial Intelligence to help screen, assess or select applicants for this position. AMD's “Responsible AI Policy” is available here.

This posting is for an existing vacancy.