Builders & open source
Stas Bekman
Develops efficient post-training software at Snowflake AI Research, Snowflake
- Develops efficient post-training software at Snowflake AI Research, Snowflake AI Research12
Stas Bekman is a software engineer who specializes in training large language and multimodal models in the PyTorch ecosystem and currently works at Snowflake AI Research on efficient post-training software. Bekman's site describes earlier work on a distributed training framework at Contextual AI and, before that, time at Hugging Face from 2020 to 2023, where Bekman was lead engineer on training the BLOOM model and later helped train IDEFICS-80b. Bekman is first author of Arctic Long Sequence Training (2025), which the paper says is open-sourced via DeepSpeed, and maintains the Machine Learning Engineering Open Book under a CC BY-SA 4.0 license.12345
At a glance
- Current role
- Develops efficient post-training software at Snowflake AI Research
- Company
- Snowflake
- Sector
- Company
- Fields
- Training systems, Language models
- Work listed
- 5
- Honors
- 1
- Sources cited
- 10
- Profile reviewed
Career
Stas Bekman has written software since 1994, has taught at technical conferences and user groups, and has published several books, including Practical mod_perl. Around 2020 Bekman helped develop fastai, then worked at Hugging Face from 2020 to 2023, contributing to the Transformers library and to projects such as DeepSpeed and PyTorch.1
After Hugging Face, Bekman worked at Contextual AI on a distributed training framework and on training and serving retrieval-augmented generation models. Bekman now works at Snowflake AI Research, developing efficient post-training software with a team that includes DeepSpeed creators.12
Large-scale training
Bekman's site names the BLOOM training run, on 384 A100 GPUs, as a major accomplishment and describes Bekman as its lead engineer; Bekman is among the co-authors of the 2022 BLOOM paper and of a companion study on choosing what language model to train with a fixed compute budget. In 2023 Bekman's team trained the IDEFICS-80b multimodal model, described in the OBELICS paper.1675
At Snowflake, Bekman is first author of the 2025 Arctic Long Sequence Training paper, which describes memory optimizations for training on multi-million-token sequences and is open-sourced through DeepSpeed. Bekman also co-authored the 2024 Universal Checkpointing paper on distributed checkpoints that can be reloaded under different parallelism settings.38
Open-source work and writing
Bekman maintains the Machine Learning Engineering Open Book, a CC BY-SA 4.0 collection of methods and tools for training, fine-tuning, and serving large models, and an early work in progress called The Art of Debugging. Bekman is also a co-author of the 2021 paper on the Hugging Face Datasets library.419
Work and contributions
Arctic Long Sequence Training (ALST) Open-source code
First author of the 2025 paper on ALST, a set of memory optimizations for training on very long sequences with Hugging Face models; the paper says ALST is open-sourced via DeepSpeed and Snowflake's Arctic Training.3
Machine Learning Engineering Open Book Published research
Author and maintainer of an open collection of methods, tools and step-by-step instructions for training, fine-tuning and serving large language and multimodal models, distributed under CC BY-SA 4.0.41
Universal Checkpointing Published research
Co-author of the 2024 paper on Universal Checkpointing, a distributed checkpointing system for large-scale training with reconfigurable parallelism.8
OBELICS and IDEFICS Open data
Equal-contribution author, with a Hugging Face affiliation, of the 2023 OBELICS paper, which releases an open interleaved image-text dataset and trains the IDEFICS models.5
BLOOM: A 176B-Parameter Open-Access Multilingual Language Model Open weights
Co-author of the 2022 BLOOM paper; Bekman's site describes serving as lead engineer for the model's training run.61
Timeline
- Began writing software, work that later spanned several technical domains.1
- Began work at Hugging Face, contributing to the Transformers library.1
- Co-authored the paper Datasets, a community library for natural language processing.9
- Co-authored a study on what language model to train with one million GPU hours.7
- Co-authored the BLOOM paper describing the 176B-parameter open-access multilingual model.6
- Co-authored the OBELICS paper, which releases an interleaved image-text dataset and IDEFICS models.5
- Co-authored the Universal Checkpointing paper on flexible distributed checkpointing.8
- Published Arctic Long Sequence Training as first author, open-sourced via DeepSpeed.3
Other documented roles
Roles the sources document that are not confirmed as current.
- Worked on a distributed training framework and on training and inference of RAG LLMs, Contextual AI1
- Worked at Hugging Face (2020–2023); lead engineer on the BLOOM training, Hugging Face15
Honors and recognition
Where they work
- Snowflake 5 people in the wiki
Related people
- Hao ZhangAlso at Snowflake
- Sridhar RamaswamyAlso at Snowflake
- Benoit DagevilleAlso at Snowflake
- Thierry CruanesAlso at Snowflake
- Torsten ScholakLanguage models · Training systems
- Naveen RaoTraining systems · Language models
- Weizhu ChenLanguage models · Training systems
- Sami JaghouarTraining systems · Language models
- Clément FarabetLanguage models · Training systems
- Lucas AtkinsLanguage models · Training systems
Suggest a correction or removal
Profiles cover public professional work only, and each statement cites a source. If something is out of date, missing or wrong, tell us and link a source. People can also ask for their profile to be corrected or removed.
References
10 sources · profile reviewed Oct 21, 2026 · data updated Oct 21, 2026
- Machine Learning: LLM/VLM Training and Engineering by Stas Bekman
- stas (Stas Bekman) on Hugging Face
- Arctic Long Sequence Training: Scalable And Efficient Training For Multi-Million Token Sequences (arXiv 2506.13996)
- stas00/ml-engineering — Machine Learning Engineering Open Book (CC BY-SA 4.0)
- OBELICS: An Open Web-Scale Filtered Dataset of Interleaved Image-Text Documents (arXiv 2306.16527)
- BLOOM: A 176B-Parameter Open-Access Multilingual Language Model (arXiv 2211.05100)
- What Language Model to Train if You Have One Million GPU Hours? (arXiv 2210.15424)
- Universal Checkpointing: A Flexible and Efficient Distributed Checkpointing System for Large-Scale DNN Training with Reconfigurable Parallelism (arXiv 2406.18820)
- Datasets: A Community Library for Natural Language Processing (arXiv 2109.02846)
- Announcing the 2023 PyTorch Contributor Awards