AI R&D Engineer – Model Training (f/m/d)

The role

You will work on the model training pipeline that was used to train Kolibri, an AI foundation model. Your work covers cluster scheduling, large-scale code execution for reinforcement learning, and evaluation code for faster inference.

What you will do

• Optimize how training loads are scheduled on the cluster
• Make the training pipeline more robust and performant
• Enable large-scale code execution for reinforcement learning
• Improve evaluation code to lift inference throughput
• Improve code maintainability through tests and design and architecture changes
• Work in mob or pair programming with the team

Required qualifications

• Ability to dive into a codebase, change its behavior, and leave it better tested and easier to understand
• Understanding of designing for low coupling and high cohesion
• Ability to refactor incrementally

Nice to have

• Experience with Python, the main language of the codebase
• Experience with Rust
• Experience with Kubernetes
• Experience with Test Driven Development, trunk based development, continuous integration and Domain Driven Design

Benefits

• 30 days of paid vacation
• Access to fitness and wellness offerings via Wellhub
• Mental health support via nilo.health
• Substantially subsidized company pension plan
• Subsidized Germany-wide transportation ticket
• Budget for additional technical equipment
• Flexible working hours and hybrid working model
• JobRad bike lease

Before you apply

• The posting does not require prior experience with a specific technology or language; it welcomes applicants at every career stage

About the company

Aleph Alpha builds foundation models in Europe for customers in finance, manufacturing and public administration. The company works from Heidelberg and fosters a culture of ownership, autonomy and extreme programming.