AI R&D Engineer – Model Training (f/m/d)
Job description
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 training pipeline's robustness, performance, and code maintainability.
At a glance
- Work mode
- Hybrid · 2 days/week in office · 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 so the team can iterate faster
- Enable large-scale code execution for reinforcement learning
- Dig into the evaluation codebase to lift inference throughput on evals
- Design high-quality tooling and infrastructure that speeds the team up long-term
- Improve maintainability of code through tests and design & architecture changes
- Identify issues and take initiative
- Make working mode, design decisions and priorities together with team mates
- Use highly collaborative techniques like mob or pair programming
What we're looking for
- Ability to dive into a codebase, change behavior as necessary, and leave the code better tested and easier to understand than before
- Understanding of how to design for low coupling and high cohesion paired with the 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 team
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 foundation model
- Customers in finance, manufacturing and public administration
- Flat organizational structure with efficient, supportive management
- Favors Extreme Programming and trunk-based development
- Often mob-program
What they value
- Ownership
- Autonomy
- Empowerment
- Open communication
- Shared purpose