Institute of Foundation Models
Open positions
Machine Learning Engineer — Pre-training (LLM)
This engineer extends and scales distributed LLM pre-training infrastructure at a foundation-model research lab: modifying frameworks like DeepSpeed/FSDP, implementing distributed optimizers from math specs, and building launch, monitoring, and checkpointing systems across multi-node GPU clusters using PyTorch or JAX.
Machine Learning Engineer — GPU Kernel
Optimizes GPU/CUDA kernel performance for foundation-model training and inference at a dedicated AI research lab: profiling deep-learning workloads, building benchmarks and automation tools, and integrating high-performance kernels with PyTorch using C++ and Python.
Machine Learning Engineer – World Model
About the Institute of Foundation Models We are a dedicated research lab for building, understanding, using, and risk-managing foundation models. Our mandate is to advance research, nurture the next generation of AI…
Research Engineer/Scientist - Speech/Audio Machine Learning
About the Institute of Foundation Models We are a dedicated research lab for building, understanding, using, and risk-managing foundation models. Our mandate is to advance research, nurture the next generation of AI…