h3About Flexion /h3pAt Flexion, we are building the autonomy stack for humanoid robots. Our mission is to drive the transition from fragile prototypes to real-world deployments of humanoids. We were 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 went from our first line of code to deploying real humanoid capabilities with our customers, leveraging simulation and reinforcement learning. Today, we are rapidly expanding the capabilities of our autonomy stack, our customer base, and our team. /ph3The role /h3pWe are looking for an experienced ML engineer to join Flexion’s experienced infrastructure team and take ownership of Flexion’s GPU compute platforms. This is a senior, on-site role with significant scope. /ppAt Flexion, we are building the brain for humanoid robots, which involves training foundation models with vast amounts of data on large GPU clusters. You will own the design, bring-up, operation and optimization of performant clusters. You will work with AI engineers to help them optimize their training speed and hardware utilization. You will also influence strategic compute planning and contribute to new tools and platforms for iterating on our AI models efficiently. This will put you at the heart of Flexion’s AI development and allow you to directly impact the execution of our ambitious roadmap. You will closely collaborate with the company’s leadership, engineers of the infrastructure team and AI engineers across the company. /ph3Key responsibilities /h3ulliArchitect, run and continuously improve existing and future cloud-based GPU clusters. Select the best frameworks and tooling to run our clusters efficiently. Work on cluster provisioning, job schedulers and monitoring systems. /liliHelp AI engineers optimize their training workloads and maximize hardware utilization using profilers, contributing to our core ML libraries. /liliContribute to short- and long-term GPU compute strategies in collaboration with our AI engineering teams and help execute on them. /liliOptimize capacity and cost by exploring multi-cloud strategies and evaluating trade-offs. /liliRaise the bar on engineering practices, including testing, code quality, documentation, and system reliability. /li /ululliDegree in Computer Science, Electrical Engineering or Software Engineering (or equivalent practical experience) plus significant industry experience. /li /ululliHands-on experience with the training or inference of large models (billions of parameters) on distributed multi-node GPU hardware. This can include bringing up and running the cluster, writing and optimizing training/inference code, building ML pipelines, etc. /liliProficiency in Python and working knowledge of PyTorch. /liliDeep understanding of distributed training concepts (DDP, FSDP, NCCL). /liliExperience with at least one cloud platform (AWS, GCP, Azure or neoclouds) or large-scale on-premises GPU infrastructure. /liliExperience with job scheduling and orchestration tools: Slurm and/or Kubernetes/KubeRay. /li /ulh3Nice-to-haves /h3ulliFamiliarity with profilers (e.g., PyTorch Profiler, Dynolog, HTA, Nsight). /liliExperience with high-performance or parallel file systems (e.g., Lustre). /liliExperience provisioning compute on multiple cloud providers. /liliExperience with infrastructure-as-code and configuration management (Terraform, Ansible). /li /ululliCompetitive Compensation /liliJoining a leading robotics team exposure to never-done-before research /liliEnergetic, collaborative culture with a bias for action and regular community events /li /ulpZurich /pulliEnhanced pension plan /liliRelocation permit sponsorship /liliEnhanced holiday paid leave perks /liliCentral Zürich office with top-tier robotics testing facilities and infrastructure /li /ulpSan Franciso /pulli401(k) with company contributions /liliHealth, dental vision coverage with the flexibility to choose your own plan /liliOpen PTO policy paid company holidays /li /ul #J-18808-Ljbffr
ML Engineer - Infrastructure Arbeitgeber: Flexion Robotics
Flexion ist ein hervorragender Arbeitgeber, der seinen Mitarbeitern die Möglichkeit bietet, an der Spitze der Robotiktechnologie zu arbeiten. Mit einem zentralen Büro in Zürich, das über erstklassige Testeinrichtungen verfügt, profitieren Sie von einer dynamischen und kollaborativen Unternehmenskultur, die Innovation und persönliche Entwicklung fördert. Unsere Mitarbeiter genießen wettbewerbsfähige Vergütung, verbesserte Rentenpläne und zahlreiche Möglichkeiten zur beruflichen Weiterentwicklung in einem der führenden Robotikteams Europas.
