Software Engineer, ML Platform

nyra health
Wien, Austria
14 days ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Job source

Tech stack

Artificial Intelligence Clinical Data Repository Cloud Computing Information Engineering Data Infrastructure Data Integrity Distributed Computing Environment Distributed Systems Job Scheduling Python (Programming Language) Open Source Technology Azure Machine Learning
+7 more
Software Engineering Management of Software Versions Data Processing Pytorch Model Validation Backend Machine Learning Operations

Job description

As a Software Engineer on the ML Platform, you will build the systems behind every nyra labs experiment and model release.

You will work across data processing, distributed training, experiment management, evaluation, inference, and release infrastructure. Your goal is to give a small research team the leverage to run ambitious experiments quickly, reproducibly, and reliably.

This is not a conventional backend role. You will work directly with researchers, understand how models are developed, and turn recurring research bottlenecks into dependable platform capabilities., * Data platform: Build reliable pipelines for ingesting, validating, transforming, versioning, and accessing large speech datasets.

  • Training infrastructure: Improve distributed training, orchestration, checkpointing, resource scheduling, and failure recovery.
  • Experiment systems: Create tooling for configuration, tracking, comparison, reproducibility, and artifact management.
  • Evaluation platform: Make it easy to run benchmarks, inspect regressions, compare releases, and understand model behavior.
  • Inference: Optimize models for efficient cloud and on-device use where relevant.
  • Release infrastructure: Automate model packaging, documentation, validation, and open-source publishing.
  • Developer experience: Build internal tools that remove friction from the daily work of researchers and engineers.
  • Reliability and security: Establish observability, access controls, and operational practices appropriate for sensitive clinical data., * Leverage-oriented: You look for improvements that make the entire team faster.
  • Reliable: You treat reproducibility and data integrity as core product requirements.
  • Self-directed: You can identify bottlenecks and own the solution end to end.
  • Collaborative: You enjoy working closely with researchers and translating experimental needs into durable systems.

Requirements

  • Strong software engineering: Excellent Python skills and experience building maintainable production systems.
  • ML systems experience: Familiarity with PyTorch training, model evaluation, GPU workloads, and the ML development lifecycle.
  • Distributed systems: Experience with cloud infrastructure, containers, orchestration, job scheduling, or distributed computing.
  • Data engineering: Experience building reliable pipelines and working with large, versioned datasets.
  • Operational mindset: You care about observability, debuggability, failure recovery, and clear system boundaries.
  • Platform thinking: You build reusable capabilities instead of solving the same problem repeatedly.
  • Research empathy: You understand that research workflows change quickly and infrastructure must support exploration.
  • AI-native workflow: You use coding agents, automation, and custom tooling to increase your own leverage and that of the team.

Benefits & conditions

  • Build the infrastructure behind open speech models and research
  • Work directly with researchers instead of operating as a separate platform team
  • Solve difficult systems problems involving large-scale audio and sensitive data
  • High ownership with visible impact on every experiment and release
  • Direct collaboration with founders and the wider nyra health engineering team
  • Attractive compensation, Phantom Stock Options, and company benefits
  • A beautiful office in Vienna’s First District with a hybrid working model

About the company

At nyra health, we build software that supports clinics, therapists, and patients throughout neurorehabilitation. myReha delivers personalized therapy, while nyra insights helps clinical teams manage and understand patient progress.

nyra labs is the research arm of nyra health. We turn difficult problems encountered in practice into open models, datasets, benchmarks, and research that the wider community can build on.

If that resonates with you, we would love to hear from you.

Apply for this position

This job is hosted externally. Click below to view the full posting and apply.

Apply on at.indeed.com

Good distractions

Talks and stories from around this role — technically off-topic, practically not.

1:52 min

Structuring and scaling the backend engineering team

Stefan Lingler Stefan Lingler +1 · Coffee With Developers

2:35 min

Preventing remote code execution in PyTorch models

Balázs Kiss · WWC 2023

1:55 min

Maintaining software and data integrity during execution

Christian Wenz Christian Wenz · WWC Europe 2026

3:03 min

Career evolution in data engineering and AI platforms

Maria Apazoglou · Coffee With Developers

1:12 min

Choosing TypeScript for complex backend applications

Maximilian Otto Maximilian Otto · WWC 2024

2:08 min

Essential engineering roles in the generative AI space

Mary Grygleski Mary Grygleski · LIVE

Videos

See all

Related articles

See all