Senior AI R&D Engineer – Model Training (f/m/d)
Level inferred from job title
Job description
Senior AI R&D Engineer in Model Training at Aleph Alpha, working across the full stack of the model training pipeline used to train Kolibri. The role focuses on improving the maintainability, robustness, and performance of research-heavy software engineering.
At a glance
- Employment
- Full-time
- Work mode
- Hybrid · Heidelberg, Germany · Berlin, Germany
What you'll do
- Work across the full stack of the model training pipeline
- Optimize how training loads are scheduled on the cluster and make the pipeline more robust and performant
- Enable large-scale code execution for reinforcement learning
- Dig deep into the evaluation codebase to lift inference throughput on evals
- Design high-quality tooling and infrastructure that speeds the team up long-term
- Identify issues and take initiative
- Improve maintainability of code through tests and design & architecture changes
- Make design decisions, working mode and priorities together with team mates
- Be on-site at least two days a week; if in Berlin, travel around every second month for a few days to Heidelberg
What we're looking for
- Ability to dive into a codebase, change behavior as necessary, and leave code better tested and easier to understand
- Understanding of how to design for low coupling and high cohesion
- Ability to incrementally refactor
- Experience with Test Driven Development (TDD)
- Experience with trunk based development
- Experience with continuous integration
- Experience with Domain Driven Design
Nice to have
- Python
- Rust
- Kubernetes
Your team
Model Training
The model training team is building and improving the model training pipeline, which has been used to train Kolibri. Most of the team is based in Heidelberg.
Benefits
- 30 days of paid vacation
- Access to a variety of fitness & wellness offerings via Wellhub
- Mental health support through nilo.health
- Substantially subsidized company pension plan
- Subsidized Germany-wide transportation ticket
- Budget for additional technical equipment
- Flexible working hours for better work-life balance and hybrid working model
- JobRad® Bike Lease
Apply
Direct apply at Alephalpha
Opens the job posting at Alephalpha
About Aleph Alpha
Our customers - in finance, manufacturing, public administration - need models that understand German, meet European regulatory requirements, and work reliably in high-stakes settings. We're building that in Heidelberg.
Mission
Aleph Alpha is one of the few companies in Europe doing serious foundation model pre-training like our recently released Kolibri-1 model.
Key facts
- Heidelberg
- Released the Kolibri-1 model
- Customers in finance, manufacturing, public administration
- Flat organizational structure
- Favor Extreme Programming and trunk-based development
- Often mob-program
What they value
- Ownership
- Autonomy
- Empowerment
- Open communication
- Shared purpose