Research Engineer, ML Infrastructure

Posted Yesterday
Be an Early Applicant
Palo Alto, CA, USA
In-Office
150K-220K Annually
Entry level
Artificial Intelligence • Machine Learning • Software
The Role
Build and own ML infrastructure for distributed training across GPU clusters, including job launching, checkpointing, recovery, monitoring, profiling, experiment tooling, and reliable data pipelines. Debug failures across GPUs, networking, numerics, and data while partnering closely with research. The role suits an early-career builder with strong Python skills, practical ML tooling experience, public work, high agency, and a willingness to learn quickly in a small, autonomous team.
Summary Generated by Built In
Who we are

Polished Snow is an AI lab building energy-based reasoning models for critical systems. Power grids, industrial infrastructure, the technology the world runs on.

Most AI models are built to generate a plausible answer. Our model, Case, is built to know when an answer is wrong before it becomes one. It scores, refines and verifies a plan against the full system until it holds. That difference matters enormously when the cost of being wrong is high.

We are a small, funded team and we are just getting started.

What we're looking for

A Research Engineer, ML Infrastructure who thinks from first principles and builds things that work outside the lab. You care about why people want something and why models behave the way they do, and you care that what we ship works when it actually matters. If you have built products, shipped software, talked to customers or run outreach, and you are drawn to the hardest reliability problems in AI, this is the role.

This is a role for someone early in their career who has already built real things. We care more about how fast you have been learning than where you have been.

What you'll do
  • Build and own distributed training systems across GPU clusters. Job launchers, checkpointing, recovery and monitoring should keep experiments moving when something fails.

  • Profile training end to end. Find bottlenecks in data loading, communication, memory and compute, then fix the ones that matter.

  • Build the tooling to launch, track and compare experiments. Make runs reproducible and shorten the path from an idea to a result.

  • Build reliable data pipelines for training and evaluation. Own data quality, throughput and the ability to reproduce a run.

  • Debug failures across GPUs, networking, numerics and data. Build systems that recover cleanly instead of leaving researchers to start over.

  • Implement and tune distributed parallelism where it helps. Understand the hardware and the tradeoffs rather than treating the training stack as a black box.

  • Work directly with research to build infrastructure before it becomes a bottleneck. You own the systems from the cluster to the training loop.

How we work

We are a small team with a high bar and very little process. You get a real problem and the trust to go after it, and you decide how it gets solved. We think carefully and move fast.

When you build something here, you own it. The system, the results, and what happens when it breaks. Nobody else is going to pick it up for you.

Who you are
  • A builder with public work to show. Repos, hardware, a product, a paper, a competition result, a project or company you ran at school. We will read it, so make it good.

  • High agency. You want to build a company from the ground up. You start things without being asked, you finish them, and you take 100% ownership of the systems you build.

  • Tapped in. You know what is happening in the tech ecosystem around you and around the world. You spend time on X, you follow what people are building, and you know and connect with the people doing it.

  • Comfortable working from first principles. You can define a problem, talk to the people who have it, and build the minimal system that proves what is possible.

  • You care how and why models work, and you care that they actually work when the stakes are real.

  • Fast to learn. Your last year looks very different from the year before it.

  • Solid in Python and at home with ML tooling. You do not have to be a researcher.

  • You thrive in small, focused teams that value autonomy, speed and tight collaboration over process.

  • You think AI for critical systems is one of the most consequential problems in the field right now and you want to be early to solving it.

  • Students and recent graduates are welcome. We do not require a degree or a set number of years of experience.

The bar

We want to work with people who want to be among the best in the world at what they do. If that sounds like you, we want to hear from you.

How to apply

Send us your best work. Tell us about the coolest thing you have built end to end and what you learned fastest in the last year. Add your X and your website if you have them.

Skills Required

  • Strong Python skills
  • Comfort with machine learning tooling
  • Experience building and shipping real projects, products, software, hardware, research, or other public work
  • Ability to work from first principles and define, validate, and build minimal systems
  • High agency, ownership, initiative, and ability to finish systems independently
  • Ability to work effectively in a small, autonomous, fast-moving team
  • Interest in how and why machine learning models work and in reliability for critical systems
  • Students and recent graduates are welcome; no degree or set number of experience years required
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
1 Employee

What We Do

Polished Snow is an AI lab building energy-based reasoning models for critical systems. The company focuses on developing artificial-intelligence technology designed to improve reasoning for essential or high-stakes applications. Its stated mission is centered on energy-based modeling and dependable system performance, although the available description does not identify specific products, customer segments, deployment environments, founding history, or additional business activities.

Similar Jobs

Chai Discovery Logo Chai Discovery

Research Engineer, ML Infrastructure

Artificial Intelligence • Biotech
In-Office
San Francisco, CA, USA
9 Employees

Anthropic Logo Anthropic

Infrastructure Engineer

Artificial Intelligence • Natural Language Processing • Generative AI
In-Office
2 Locations
2500 Employees
350K-500K Annually

Cognition Logo Cognition

Research Engineer, ML Infrastructure

Artificial Intelligence • Software
In-Office
San Francisco, CA, USA
49 Employees

Hilton Logo Hilton

Food and Beverage Supervisor - Hilton San Diego Bayfront

Events • Food • Travel • Hospitality • Hotels & Resorts
In-Office
San Diego, CA, USA
500000 Employees

Similar Companies Hiring

Revel Thumbnail
Aerospace • Hardware • Robotics • Software
Marina Del Rey, California
70 Employees
Blee Thumbnail
Artificial Intelligence • Marketing Tech • Software
New York, New York
33 Employees
Vega Thumbnail
Artificial Intelligence • Automotive • Insurance • Transportation
US
65 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account