Foundation Model Pretraining Engineer (End-to-End)

Brahma Consulting Group; San Francisco (CA); On-site; USD 150,000 - 210,000; Full time

Job summary

Brahma Consulting Group in San Francisco seeks a Research Engineer for Foundation Models who will train models from scratch and own pretraining work end-to-end.

The role emphasizes architecture decisions, data pipelines, and running training runs, using PyTorch and JAX in a distributed multi-node setting.

This early-stage lab values hands-on execution and projects with practical transfer to physical systems.

Qualifications

• Experience training models from scratch, not only fine-tuning.
• Ownership of end-to-end training runs and debugging training runs.
• Hands-on work with data infrastructure enabling large-scale runs.

Responsibilities

• Pretrain foundation models for physical systems with distributed multi-node setups.
• Make architecture decisions, design data pipelines, and run training runs.
• Contribute to a small, fast-moving team focused on impactful results.

Skills

End-to-end training

Distributed training

Data pipeline development

Tools

PyTorch

JAX

Job description

Brahma Consulting Group in San Francisco seeks a Research Engineer for Foundation Models who will train models from scratch and own pretraining work end-to-end.

The role emphasizes architecture decisions, data pipelines, and running training runs, using PyTorch and JAX in a distributed multi-node setting.

This early-stage lab values hands-on execution and projects with practical transfer to physical systems.