Prior Labs Logo

Prior Labs

Infrastructure Engineer

Reposted 2 Months Ago
Be an Early Applicant
In-Office
New York, NY, USA
Senior level
In-Office
New York, NY, USA
Senior level
The Senior ML Infrastructure Engineer will own and optimize GPU infrastructure for training tabular foundation models, ensuring high efficiency and collaboration with research teams.
The summary above was generated by AI
Who we are

Foundation models transformed text and images. Structured data - the largest and most consequential data format in the world - stayed untouched, until now. What LLMs did for language, we're doing for tables.

We pioneered tabular foundation models: TabPFN v2 was a Nature cover story, has passed 3.5M+ downloads and 7,500+ GitHub stars, and runs in production from detecting lung disease with Oxford Cancer Analytics to preventing train failures with Hitachi. The hardest problems - millions of rows, real-time inference, entirely new modalities - are still open, and no one else is working on them at this level.

We're a small, highly selective team of 40+ with backgrounds from Google, DeepMind, Meta, Apple, Amazon, Jane Street, and CERN, led by Frank Hutter, Noah Hollmann, and Sauraj Gambhir, and advised by Bernhard Schölkopf and Turing Award winner Yann LeCun.

In July 2026, less than 18 months after our €9M pre-seed, we joined SAP as an independent frontier AI lab - same team, mission, and open-weights models, now backed by more than €1 billion over four years.

About the Role

We spend tens of millions per year on GPU compute to train tabular foundation models. That's not a target, it's what we're running today, and it's growing. The person who owns this infrastructure makes decisions worth millions of dollars: cluster architecture, scheduling efficiency, provider strategy, hardware selection. A wrong call costs six figures.

Today we run on Slurm across multiple clusters. We're scaling to multi-cluster, multi-provider infrastructure using SkyPilot and evaluating new hardware generations as they come online. You own the full stack, from cluster orchestration and cost optimization to CI/CD, automation, and the tooling that keeps researchers moving fast. You work directly with the research team and understand what they're doing well enough to make infrastructure decisions that actually help them.

What you'll work on:

  • Own and evolve multi-cluster GPU infrastructure, from Slurm and SkyPilot on GCP today to multi-provider orchestration and new GPU generations tomorrow. Drive architecture, automated provisioning, workload scheduling, reliability, cost optimization, and capacity planning against growing compute demands

  • Build the developer productivity layer: CI/CD pipelines, automated testing, reproducible environments, experiment tracking, and internal tooling that keeps research iteration speed high

  • Improve infrastructure reliability through monitoring, observability, infrastructure as code, and automation. Reduce operational toil and make complex systems easier to manage.

  • Own the compute budget. You understand cost per FLOP across providers and hardware, and you hate wasted compute

Tech stack: Slurm, SkyPilot, GCP (+ AWS and Azure to come), Docker, Terraform, GitHub Actions, Python, uv, wandb

You may be a good fit if you have:

  • 3+ years building and operating production infrastructure at scale, ideally involving compute clusters, distributed training systems, or HPC

  • Hands-on experience with cluster management and workload orchestration systems such as Slurm, Kubernetes, or similar technologies. You've debugged scheduling failures and operated infrastructure where downtime has real cost

  • Experience designing and maintaining CI/CD pipelines, automated testing, and reliable build and deployment workflows

  • Strong systems engineering fundamentals: Linux, networking, storage, and distributed systems. You're comfortable debugging complex infrastructure problems across the stack.

  • Strong Python skills and enough familiarity with training to profile a run and tell whether the bottleneck is data loading, communication, or compute

  • Track record of making infrastructure decisions that measurably improved reliability, developer productivity, or cost efficiency

  • Strong AI tooling skills. You use Claude Code, Cursor, or similar fluently to move fast without sacrificing quality

Bonus:

  • Familiarity with GPU infrastructure, PyTorch, GPU profiling, and distributed training

  • Experience operating at tens-of-millions-scale compute spend

  • Multi-cloud or hybrid HPC/cloud infrastructure experience

  • Experience scaling from single cluster to multi-cluster orchestration

  • Background building experiment tracking, model registries, or ML pipeline tooling

Life at Prior Labs

You'll work alongside researchers and builders who hold themselves to a very high bar - in the quality of their work and in how they work with each other. We move fast and still take the time to do things right.

Our teams are based in Berlin, Freiburg, and New York - when you're working on something as hard as TabPFN, being in the same room matters. But great people come from everywhere, and in exceptional cases we're open to remote, which usually means frequent travel to one of our offices. Wherever you're based, the whole company comes together regularly for offsites to build and celebrate together.

Our Commitments

The best products and teams are built by people with a wide range of perspectives and backgrounds. We welcome applications from all identities and walks of life - especially if you've ever felt discouraged by "not checking every box" - and provide equal opportunities regardless of gender, sexual orientation, origin, disability, or any other trait that makes you who you are.

We care about how your data is handled - see our Recruiting Data Privacy page

Similar Jobs

7 Days Ago
Hybrid
60K-70K Annually
Senior level
60K-70K Annually
Senior level
Fintech • HR Tech • Payments • Software
Own and scale AWS infrastructure supporting high-availability, low-latency fintech APIs across multiple regions. Manage Kubernetes clusters, CI/CD pipelines, ELK observability, infrastructure security, monitoring, incident response, and infrastructure automation. The role involves optimizing distributed systems, supporting compliance, enabling frequent deployments, and participating in on-call operations within a fast-moving startup environment.
Top Skills: AWSBashCloudFormationCloudwatchDockerEc2EcsEksElasticsearchElk StackGitlab Ci/CdIamKibanaKubernetesLambdaLogstashPythonRdsTerraformVpc
7 Days Ago
In-Office or Remote
70K-80K Annually
Senior level
70K-80K Annually
Senior level
Software • Analytics
Architect and scale secure, reliable cloud infrastructure for a globally used SaaS platform. Own infrastructure strategy, uptime, SLAs, incident management, observability, Infrastructure as Code governance, CI/CD reliability, cost optimization, and engineering enablement. The role requires deep AWS and cloud architecture expertise, production systems ownership, and the ability to influence technical direction across engineering, product, and security.
Top Skills: AlertingAWSCi/CdCloud InfrastructureInfrastructure As CodeKubernetesMonitoringObservability
7 Days Ago
Hybrid
55K-75K Annually
Junior
55K-75K Annually
Junior
Information Technology • Social Impact • Software • Consulting
Supports global infrastructure and security operations for a revenue-critical SaaS platform. Responsibilities include managing Windows and cloud systems, Kubernetes migration, Grafana observability, Cloudflare services, Microsoft 365 administration, identity and endpoint security, vulnerability management, compliance controls, incident response, documentation, automation, and on-call support. The role also contributes to infrastructure-as-code, containerisation, disaster recovery, and continuous improvement across a distributed, multi-region environment.
Top Skills: AnsibleAsp.Net CoreBitdefender GravityzoneBitwardenC#Ci/CdCloudflareContainerdDockerDynamicsEntra IdExchange OnlineGitopsGoGrafana OssHelmImis RiseJavaScriptKnowbe4KubectlKubernetesLokiMicrosoft 365MongoDBPowershellPrometheusPulumiPythonRedisSalesforceSharepointSQL ServerTerraformTypescriptWindows Server 2022

What you need to know about the NYC Tech Scene

As the undisputed financial capital of the world, New York City is an epicenter of startup funding activity. The city has a thriving fintech scene and is a major player in verticals ranging from AI to biotech, cybersecurity and digital media. It also has universities like NYU, Columbia and Cornell Tech attracting students and researchers from across the globe, providing the ecosystem with a constant influx of world-class talent. And its East Coast location and three international airports make it a perfect spot for European companies establishing a foothold in the United States.

Key Facts About NYC Tech

  • Number of Tech Workers: 549,200; 6% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Capgemini, Bloomberg, IBM, Spotify
  • Key Industries: Artificial intelligence, Fintech
  • Funding Landscape: $25.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Greycroft, Thrive Capital, Union Square Ventures, FirstMark Capital, Tiger Global Management, Tribeca Venture Partners, Insight Partners, Two Sigma Ventures
  • Research Centers and Universities: Columbia University, New York University, Fordham University, CUNY, AI Now Institute, Flatiron Institute, C.N. Yang Institute for Theoretical Physics, NASA Space Radiation Laboratory

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account