20 beste Stellen als machine learning python in kanton zürich – aktuell auf Personalsuche

At Flexion, we're building the intelligence layer powering the next generation of humanoid robots. Our mission is to accelerate the transition from fragile prototypes to real-world humanoid deployment. We are founded by leading scientists in robot reinforcement learning (ex-Nvidia, ex-ETH Zürich), and backed by leading international VC firms. In just months, we’ve gone from our first line of code to deploying real humanoid capabilities.

The Role

This is a senior ML engineering, building out our core compute and data platforms. We’re building the brain for humanoid robots, which involves training large-scale foundational models with vast amounts of data. You'll architect the pipelines that move data from simulators and robots into model training, optimize training workloads and create the platforms that help our AI engineers train, evaluate, and iterate fast.

You'll join Flexion's experienced Infrastructure team (ex-Google, Meta, Amazon) and take significant ownership of the systems behind our data collection, training and experimentation workflows: from strategic infrastructure decisions, cluster orchestration and distributed training optimization to data platforms, CI, and experiment tooling. This is a senior, on-site role at our Zürich office.

Key Responsibilities

Architect training data platforms and pipelines: build the storage, processing, and serving layers that handle the full data lifecycle: from simulator output and robot telemetry to training datasets. This includes building infrastructure with object storage (S3), parallel filesystems (Lustre), and common data formats (Parquet, WebDataset, LeRobot). Use distributed processing frameworks (Ray, Spark) to transform and validate data at scale.
Optimize distributed training: work with our AI engineers to scale workloads across multi-node GPU clusters, profiling and improving throughput, device utilization, and communication efficiency. This includes optimizing our distributed IsaacLab-based sim-to-real training.
Evaluate and adopt new platforms and technologies: compare cloud providers, GPUaaS platforms, and emerging tooling, owning the decisions on what we adopt as we grow our compute footprint.

Requirements

Professional experience building infrastructure, tooling, or platforms for large-scale ML training workloads.
Strong experience with ML data infrastructure: distributed processing, object storage, metadata/catalog systems, dataset versioning, streaming, shuffling, caching, and high-throughput dataloading.
Hands-on experience training, scaling, or supporting distributed ML workloads, with understanding of DDP, FSDP, NCCL, checkpointing, fault tolerance, and training performance bottlenecks.
Experience with cloud infrastructure such as AWS, GCP, or similar, including compute, networking, storage, and cost/performance tradeoffs.
Experience with job scheduling or orchestration systems such as Slurm, Kubernetes, Ray, or similar.
Proficiency in Python and working knowledge of PyTorch.
Ownership mindset: comfortable making architectural decisions, setting direction, and delivering independently in a fast-moving environment.

Nice to have

Experience with job scheduling and orchestration tools: Slurm, Kubernetes, or both.
Familiarity with common data formats (Parquet, WebDataset, LeRobot).
Familiarity with robotics simulation environments (IsaacLab, IsaacGym, MuJoCo).
Experience with infrastructure-as-code and configuration management (Terraform, Ansible).
Familiarity with experiment tracking platforms (Weights & Biases, MLflow).

Benefits

Competitive compensation package
A front-row seat at one of Europe’s most ambitious robotics companies
An energetic, collaborative team with a bias for action

Direkt bewerben

Erweiterte Suche

machine learning python Jobs in kanton zürich

Machine Learning Engineer

Internship in Machine Learning (m/f) - 50-100%

Research assistant positions

Robotics Software Intern - Sim-to-Real

Applied AI Architect, Industries

AI & Data Science Intern - Finance Transformation (m/f/x/d, 60-100%), Zurich (Zurich, CH)

Software Engineer III, AI/ML, DSP/HTP Optimisation

Data Scientist (m/f/d)

Data Scientist Analytics Non-Financial Risks (w/m/d)

Software Engineer III, AI/ML Shopping Creator, YouTube

Enterprise-AI Application Developer

PhD Position on Sleep, Stress and Arousal in Humans

AI CAD CAM IoT Engineer / Tinkerer CNC

PhD in Network Science: Modeling of Animal Networks

AI Research Engineer - Dexterous Manipulation (Egocentric Models)

Software Engineer III, 3D Modelling, XR

PhD Position in Vehicle Sensor and Remote Sensing Analysis for Road Safety

Marie Skłodowska-Curie Doctoral Training Network – Coupled Problems for Decarbonization in Industry and Power Generation (COMBINE)

Member of Technical Staff (Backend & Infrastructure)

Internship - Humanoid Motion Generation (Diffusion or Flow Matching)

Ich möchte aktuelle Job-E-Mails für Stellen als machine learning python in kanton zürich erhalten

Related Searches

Weitere Job-Standorte erkunden

Tools für Jobsuchende

Arbeitgebertools

Durchsuchen

Bleiben Sie in Kontakt