Skip to content
New Job?Nejo!

The AI Job Search Engine

HEHelsing

AI Research Engineer - ML Engineering(m/w/x)

Berlin, München
Full-timeOn-siteExperienced
AI/ML
Data Science

Scaling deep learning frameworks on PyTorch for distributed training and GPU optimization in defence AI. MSc/PhD in CS/STEM and strong Python software engineering skills required. Stock options, relocation support, and social allowances.

Requirements

  • MSc or PhD in Computer Science or STEM
  • Strong software engineering skills in Python
  • Fluency with modern DL frameworks
  • Ability to write custom layers and loops
  • Clear communication and theoretical concept building
  • First-principles mindset and rapid code integration
  • Experience debugging production ML pipelines
  • Experience training models on GPU clusters
  • Knowledge of cross-node communication patterns
  • Experience with large-scale multi-modal datasets
  • Proficiency with workload orchestrators
  • Low-level understanding of GPU architecture

Tasks

  • Extend integrated deep learning frameworks built on PyTorch
  • Enhance framework efficiency and ease of use
  • Scale infrastructure and tooling for distributed training
  • Design data strategies for large-scale datasets
  • Optimize storage to ensure maximum GPU utilization
  • Architect tools and platforms for autonomous decision-making
  • Abstract complex distributed systems to increase throughput
  • Maximize developer velocity through robust platform engineering

Work Experience

approx. 1 - 4 years

Education

Master's degree

Languages

EnglishBusiness Fluent

Tools & Technologies

PythonPyTorchJAXTensorFlowNCCLMPISlurmKubernetesRayGPU

Benefits

Informal Culture

  • Outcome-focused work culture

Competitive Pay

  • Competitive compensation
  • Stock options

Other Benefits

  • Relocation support

Additional Allowances

  • Social allowances
  • Education allowances

Team Events

  • Regular company events
  • Regular all-hands meetings

Mentorship & Coaching

  • Hands-on onboarding program
Find the original job posting in its most current version here. Nejo automatically captured this job from the website of Helsing and processed the information on Nejo with the help of AI for you. Despite careful analysis, some information may be incomplete or inaccurate. Please always verify all details in the original posting! Content and copyrights of the original posting belong to the advertising company.
Not a perfect match?
100+ Similar Jobs for you
  • Helsing

    AI Research Engineer - Foundation Models(m/w/x)

    Full-timeOn-siteExperienced
    Berlin, München
  • Prior Labs

    ML Engineer, Foundation Model(m/w/x)

    Full-timeOn-siteExperienced
    from 120,000 / year
    Berlin, Freiburg im Breisgau
  • Prior Labs

    MLOps / ML Systems Engineer(m/w/x)

    Full-timeOn-siteSenior
    Berlin, Freiburg im Breisgau
  • Prior Labs

    ML Engineer, Cloud Platform(m/w/x)

    Full-timeOn-siteExperienced
    from 140,000 / year
    Berlin, Freiburg im Breisgau
  • Perplexity

    Machine Learning Research Engineer - Search(m/w/x)

    Full-timeOn-siteExperienced
    Berlin
100+ View all similar jobs

Nejo is an AI – results may be incomplete or contain mistakes