Pre-training Research Engineer
Sciforium · San Francisco
Sciforium is an AI infrastructure company developing next-generation multimodal AI models and a proprietary, high-efficiency serving platform. Backed by multi-million-dollar funding and direct sponsorship from AMD with hands-on support from AMD engineers the team is scaling rapidly to build the full stack powering frontier AI models and real-time applications.
About the role
As a Pre-training Research Engineer, you’ll focus on model implementation, pertaining and scaling, and improving the quality of our byte-native and multimodal foundation models. You’ll build and iterate quickly on research ideas, contribute production-grade training code and infrastructure, and help deliver high-quality base models that can serve real-world use cases at scale.
Key Responsibilities
Pre-training & Scaling
Train large byte-native and multimodal foundation models across massive, heterogeneous corpora.
Implement and evaluate new model architectures, training objectives, and optimization methods.
Develop stable pre-training recipes and run scaling experiments for novel architectures.
Conduct ablations and analyze training dynamics, model behavior, and base-model quality.
Work with data and distributed training engineers to improve training efficiency, reliability, and scalability.
Must-Haves
5+ years of experience in machine learning research or engineering, with a proven track record of developing and pre-training large language or multimodal foundation models.
Software Engineering: Strong general software engineering skills, with the ability to write robust and performant training code.
ML Foundations: Solid understanding of deep learning fundamentals and modern pre-training methods and literature.
Research and Experimentation: Ability to quickly implement research ideas and evaluate them using clear baselines, ablations, metrics, and analysis.
GPU and Distributed Training: Hands-on experience running training workloads in GPU-based environments, with familiarity with distributed training.
Education: MS in Computer Science, Machine Learning, Artificial Intelligence, Mathematics, or a related field.
Nice-to-Haves
PhD in Computer Science, Machine Learning, Artificial Intelligence, Mathematics, or a related field.
JAX Ecosystem: Extensive experience with the JAX, Flax, and XLA stack.
Large-Scale Distributed Training: Experience with multi-node pre-training using systems such as FSDP, ZeRO, or Megatron.
Training Recipes and Scaling: Experience developing training recipes, ablations, or scaling experiments.
Monitoring and Reproducibility: Experience owning end-to-end training and evaluation pipelines with monitoring and reproducibility.
Education
MS or PhD in Computer Science, Machine Learning, Artificial Intelligence, Mathematics, or a related field.
Benefits include
Medical, dental, and vision insurance
401k plan
Daily lunch, snacks, and beverages
Flexible time off
Competitive salary and equity
Equal opportunity
Sciforium is an equal opportunity employer. All applicants will be considered for employment without attention to race, color, religion, sex, sexual orientation, gender identity, national origin, veteran or disability status.
More jobs at Sciforium · Software Engineering jobs in San Francisco