Research Engineer, Foundation Models

Research Engineer, Foundation Models. Early-stage lab in SF, well funded, small team. You'd train models from scratch rather than adapt existing ones.

Client is building foundation models for physical systems. The team is small enough that you'd own real surface area, and the work is pretraining: architecture decisions, data pipeline, and the training runs themselves.

You've been close to model training, not just model usage. Maybe you've owned a run end to end, maybe you've been the person debugging a loss curve at 2am, maybe you've built the data infrastructure that made a run possible. We're more interested in what you've touched than in which frameworks you touched it with.

Stack is PyTorch and JAX, distributed training across multi-node setups. If you've worked in that world, great. If you've done equivalent work in a different stack, we'd still like to talk.

Worth knowing: this is a pretraining role. If your experience is primarily fine-tuning or adapting released models, it's probably not the right fit, though we'd still encourage you to apply if you think the underlying skills transfer.

Broader than the req warrants, on purpose. These feed Linked In's recommendation engine more than they feed candidate screening.