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.What we care about: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.Skills tagsPyTorch, JAX, Distributed Systems, Deep Learning, Machine Learning, Model Training, Python, CUDA, Research, Computer VisionBroader than the req warrants, on purpose. These feed LinkedIn's recommendation engine more than they feed candidate screening.