Olmo-core
PyTorch building blocks for the OLMo ecosystem
...It underpins the broader OLMo ecosystem and includes components for models, data loading, distributed execution, optimization, checkpointing, evaluation, and generation. Official scripts reproduce training configurations for released OLMo models. The library supports advanced attention backends, mixture-of-experts architectures, float8 training, and fused low-memory operations through optional dependencies. Jobs can run with standard torchrun or AI2’s Beaker infrastructure. Direct autoregressive generation is also available for experimentation and chat-style inference. ...