Artificial Intelligence Engineer (f/m/d) Dual Study Conversion

About the team:

We are part of the HPE HPC & AI Go-to-Market organization for EMEA, responsible for bringing supercomputers and AI factories to market and ensuring customer success. Our Center of Excellence for Supercomputers & AI Factories is a compact team of highly skilled experts who collaborate closely with customers to maximize system performance, optimize workflows, and drive joint research and engineering projects.

The role:

You will work with the High-Performance Computing Center Stuttgart (HLRS) to support the deployment and adoption of their HammerHAI factory being delivered this year. The role requires two days per week office presence at HRLS.

As Artificial Intelligence (AI) Engineer, you will:

• Design, implement, and optimize AI/ML components and pipelines for scientific workloads running on HPC and hybrid cloud platforms.
• Work with HLRS researchers and HPE teams to productise models, accelerate workflows on accelerated hardware, and co-develop scalable AI solutions for HammerHAI use cases.

Key responsibilities are:

• Design and develop AI/ML models, data pipelines, and inference/training workflows tailored to scientific and engineering applications.
• Optimize deep-learning frameworks (e.g., PyTorch, TensorFlow, JAX) and models for GPU/accelerator-based HPC environments, including performance tuning, mixed-precision, and distributed training.
• Translate research prototypes into robust, maintainable code and containerized deployments; contribute enhancements and integrations for subsystems of end-user applications.
• Create and execute testing, benchmarking, and validation plans for AI workloads; identify, debug, and remediate performance or correctness issues.
• Collaborate as a core member of cross-functional project teams (HLRS researchers, HPE engineers, and external partners) to deliver cost effective, high-quality AI solutions and share best practices.
• Produce technical documentation, run knowledge-transfer sessions, and communicate project status and design proposals to stakeholders.

If you have these Skills, Expertise, Attitude and Education…

• Bachelor’s or Master’s degree in Computer Science, Data Science, Machine Learning, or a related field (or equivalent experience)
• 2–5 years’ experience applying AI/ML in a scientific or HPC environment; hands-on work with large-scale model training and inference
• Proficiency with deep-learning frameworks (PyTorch, TensorFlow, JAX), HPC toolchains (MPI, Slurm), containerization (Docker, Singularity), and model optimization libraries (e.g., ONNX, NVIDIA/TensorRT)
• Strong analytical and problem-solving skills, with experience in performance profiling, debugging, and optimization on accelerators
• Familiarity with software engineering best practices: version control, CI/CD, testing methodologies, and reproducible workflows
• Excellent written and verbal communication skills in English; German proficiency desirable
• Collaborative mindset, ability to present technical proposals and negotiate implementation options across multi-disciplinary teams
• Optional / nice-to-have: Contributions to open-source ML/HPC projects or published research in AI/HPC

Part-time work or job-sharing is also applicable to this position.

What We Can Offer You:

Health & Wellbeing

We strive to provide our team members and their loved ones with a comprehensive suite of benefits that supports their physical, financial and emotional wellbeing.

Personal & Professional Development

We also invest in your career because the better you are, the better we all are. We have specific programs catered to helping you reach any career goals you have — whether you want to become a knowledge expert in your field or apply your skills to another division.

Unconditional Inclusion

We are unconditionally inclusive in the way we work and celebrate individual uniqueness. We know varied backgrounds are valued and succeed here. We have the flexibility to manage our work and personal needs. We make bold moves, together, and are a force for good.