Logo Aleph Alpha

Senior AI Researcher

Job

  • Level
    Senior
  • Job Feld
    Data
  • Anstellung
    Vollzeit
  • Vertragsart
    Unbefristetes Dienstverhältnis
  • Ort
    Heidelberg
  • Arbeitsmodell
    Hybrid, Onsite
  • Job Zusammenfassung

    In dieser Rolle entwickelst du innovative Verfahren im Bereich Reinforcement Learning, führst umfangreiche Experimente durch und optimierst Schulungsinfrastrukturen, um die Leistung unserer Modelle zu steigern.

    Job Technologien

    Deine Rolle im Team

    • Aleph Alpha is one of the few companies in Europe with end-to-end in-house model development including pre- and post-training. We're building models that have general-purpose capabilities, but also specifically excel at addressing the needs of our customers.
    • We're growing our post-training team in Heidelberg (or hybrid in Germany) and are looking for an AI Researcher who combines a deep theoretical understanding of reinforcement learning methods with a desire to improve on the state of the art and improve model capabilities in large-scale training.
    • As a (senior) AI Researcher for reinforcement learning you will shape and improve the underlying RL methodology, maintain a high-quality training code-base, and conduct large-scale experiments to hill-climb our performance benchmarks.
    • In your day-to-day you will conduct large-scale reinforcement learning experiments, derive hypotheses from the results, and iterate on both the implementation and methodology based on the observations.
    • Together with a collaborative team, you will have direct impact on the models that we ship to our customers.
    • Hill-climb in large-scale training: Conduct large-scale LLM training runs, analyze evaluation scores in depth, propose hypotheses for improvement and directly implement them in order to maximize performance on our benchmarks.
    • Theoretical innovation: Stay at the bleeding edge of RL research. You will identify, implement, and iterate on novel approaches to multi-turn reinforcement learning.
    • Scale our training infrastructure: Identify bottlenecks in our training setup and optimize our RL training loops for large-scale training.
    • Cross-functional collaboration: Partner with our other post-training teams to turn raw feedback into actionable training signals, ensuring that our RL iterations lead to measurable improvements in downstream performance.

    Unsere Erwartungen an dich

    Qualifikationen

    • A deep understanding of Reinforcement Learning theory and how it relates to modern RL methods.
    • Familiarity with statistical methods for evaluation and experiment design.
    • Ability to reason about what an evaluation/environment measures and whether it matters - not just run benchmarks, but understand them.
    • Strong Python skills and comfort with ML tooling (especially torch distributed).
    • Willingness to relocate to Heidelberg or travel regularly (potentially weekly).
    • A history of contributions to top-tier venues (NeurIPS, ICML, ICLR, etc.) specifically regarding RL.

    Erfahrung

    • Experience with multi-node LLM training (ideally using RL). You understand how to scale multi-node RL trainings and can reason about and implement distributed algorithms.
    • PhD in reinforcement learning or equivalent research experience.
    • Experience evaluating LLM models and crafting environments for training.

    Benefits

    Gesundheit, Fitness & Fun

    Themen mit denen du dich im Job beschäftigst

    Job Standorte

    • Standort Heidelberg

      Baden-Württemberg

      Deutschland

    Das ist dein Arbeitgeber

    Aleph Alpha

    Aleph Alpha

    Als deutsches KI-Startup mit Sitz in Heidelberg fokussiert sich Aleph Alpha auf die Entwicklung von großen Sprachmodellen und generativer KI. Es bietet Lösungen für Unternehmen, die ihre eigenen KI-Kompetenzen aufbauen und ihre Daten schützen möchten.

    Description

  • Unternehmenstyp
    Startup
  • Arbeitsmodell
    Hybrid, Onsite
  • Branche
    Internet, IT, Telekom
  • Logo Aleph Alpha

    Senior AI Researcher

    Ort
    Heidelberg
    Arbeitsmodell
    Hybrid, Onsite
    Diversität
    Für alle Personen geeignet (m/w/d)
    Nur Englisch
    Nur Englisch erforderlich

    Weitere Jobs