Omnifold Logo

Omnifold

Infrastructure Tech Lead

Posted Yesterday
Be an Early Applicant
In-Office
San Francisco, CA, USA
Expert/Leader
In-Office
San Francisco, CA, USA
Expert/Leader
Own infrastructure for deploying and operating customer-specific AI models and services. Responsibilities include AWS cloud and GPU resource management, CI/CD and infrastructure as code, customer data isolation and security, SOC 2 compliance, monitoring and logging, ETL pipelines, model versioning, ML lifecycle management, and automated testing. The role requires strong Python skills, infrastructure or platform engineering experience, startup experience, and technical leadership.
The summary above was generated by AI
Infrastructure Tech Lead / Principal Engineer

Omnifold trains custom AI models that help planners forecast the future. We are hiring our first infrastructure tech lead, who will own the systems that make everything else possible.

What makes this job interesting:

  • We train a unique model for each customer, which means model training and inference work differently here than at any other company. You’ll never get more reps building model training infrastructure!

  • Our team has very fast iteration speed but needs robust monitoring to pick up signal on user patterns. This is especially important as our application interface for AI-driven forecasting is unique on the market.

What you’ll own

  • Deployment: Reliable processes for getting models and services into production

  • Security: Data isolation between customers, product security, infrastructure hardening (SOC2 compliance and beyond)

  • Cloud resource management: GPU allocation, instance sizing, cost optimization

  • Monitoring and logging: Visibility into what's running, what's failing, and why

  • Data and ML ops: ETL pipelines from varied customer data sources, model versioning and lifecycle management

  • Automated testing: Building the test infrastructure that lets us ship with confidence

What we’re looking for

  • Experience with cloud computing (especially GPU workloads), CI/CD infrastructure-as-code. We run on AWS

  • Familiarity with or interest in ML workflows

  • Security fundamentals: encryption, access controls, compliance basics

  • Python proficiency

  • Ideally ~10 years of experience, including startup experience, with at least 3 years in a tech lead role. 5+ years in infrastructure, DevOps, or platform engineering roles

  • Must have a strong Computer Science background

Location: San Francisco (in-person, 5 days per week)

Omnifold’s Mission

Every bad forecast has a physical consequence. Unnecessary goods are manufactured, shipped, and stored. Emergency air freight is needed for misallocated products. Poor production planning means workers show up with nothing to do, or work frantic overtime. Inefficiency is everywhere.

Our mission is to eliminate waste and accelerate growth for every company with physical products.

Similar Jobs

7 Days Ago
In-Office or Remote
225K-325K Annually
Senior level
225K-325K Annually
Senior level
Artificial Intelligence • App development
Lead the GPU cluster infrastructure team while remaining at least 50% hands-on. Own technical direction, architecture, scheduling, storage, security, infrastructure as code, observability, incident response, and platform roadmap. Manage hiring, coaching, project prioritization, and provider and researcher escalations. Operate production Kubernetes GPU clusters across multiple providers and improve performance, reliability, fault tolerance, and workload isolation for large-scale AI research.
Top Skills: AnsibleArgocdC++CephDcgmFirecrackerGoGpusGvisorHelmInfinibandKataKubernetesKueueLinuxLustreNcclNvidia Gpu OperatorPrometheusPythonPyTorchRbacRoceRustSlurmTerraformVastVolcanoWeka
One Month Ago
In-Office
San Francisco, CA, USA
200K-325K Annually
Senior level
200K-325K Annually
Senior level
Artificial Intelligence • Software • Database
Lead and build Reducto's ML infrastructure: own roadmap, implement and optimize training and inference stacks, improve multi-node/multi-GPU performance, GPU utilization, latency, throughput, reliability and observability, develop benchmarks, and enable fast experiment-to-production workflows while mentoring engineers.
Top Skills: CudaDistributed TrainingGpuKubernetesPythonPyTorchRaySglangTensorrt-LlmTritonVllm
One Month Ago
In-Office
155K-180K Annually
Senior level
155K-180K Annually
Senior level
Artificial Intelligence • Machine Learning
The Portal & Infrastructure Tech Lead will be responsible for the technical architecture and engineering of the FF EAI Brain Developer Open Platform, focusing on API design, SDK engineering, and leading the engineering team while ensuring platform quality and community engagement.
Top Skills: GrpcOauth 2.0Restful ApisRobotics Software ArchitecturesRosRos2Sdk

What you need to know about the San Francisco Tech Scene

San Francisco and the surrounding Bay Area attracts more startup funding than any other region in the world. Home to Stanford University and UC Berkeley, leading VC firms and several of the world’s most valuable companies, the Bay Area is the place to go for anyone looking to make it big in the tech industry. That said, San Francisco has a lot to offer beyond technology thanks to a thriving art and music scene, excellent food and a short drive to several of the country’s most beautiful recreational areas.

Key Facts About San Francisco Tech

  • Number of Tech Workers: 365,500; 13.9% of overall workforce (2024 CompTIA survey)
  • Major Tech Employers: Google, Apple, Salesforce, Meta
  • Key Industries: Artificial intelligence, cloud computing, fintech, consumer technology, software
  • Funding Landscape: $50.5 billion in venture capital funding in 2024 (Pitchbook)
  • Notable Investors: Sequoia Capital, Andreessen Horowitz, Bessemer Venture Partners, Greylock Partners, Khosla Ventures, Kleiner Perkins
  • Research Centers and Universities: Stanford University; University of California, Berkeley; University of San Francisco; Santa Clara University; Ames Research Center; Center for AI Safety; California Institute for Regenerative Medicine

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account