Skip to content
AI News HubLIVE
Original source2 min read

OpenLanguageModel: Readable and Composable Small-Language-Model Pretraining for Education and Research

Summary

OpenLanguageModel (OLM) is an open-source PyTorch library for building and pretraining small language models with visible internals. Model code mirrors architecture using components like Block, Residual, Repeat, and Parallel. It integrates tokenizers, datasets, optimization, mixed precision, callbacks, checkpoints, and hardware-aware execution, enabling seamless transition from teaching notebooks to full pretraining. The library includes 27 presets across nine model families and documentation from fundamentals to research. Validation shows close agreement with reference implementations, 90.6% weak-scaling efficiency on four GPUs for a 348M-parameter model, and positive usability feedback. OLM is MIT-licensed and available on PyPI, GitHub, and its documentation site.

SourcearXiv Computational LinguisticsAuthor: Tavish Mankash, Vardhaman Kalloli, Keshava Prasad, Deepan Muthirayan
OpenLanguageModel: Readable and Composable Small-Language-Model Pretraining for Education and Research
Report an error

The correction channel is not available yet. You can copy the article reference below for later.

Correction instructions
Read article

[Submitted on 18 Jul 2026]

Title:OpenLanguageModel: Readable and Composable Small-Language-Model Pretraining for Education and Research

View a PDF of the paper titled OpenLanguageModel: Readable and Composable Small-Language-Model Pretraining for Education and Research, by Tavish Mankash and 3 other authors

View PDF HTML (experimental)

Abstract:OpenLanguageModel (OLM) is an open-source PyTorch library for building and pretraining small language models while keeping their machinery visible. In OLM, model code reads like the architecture: components are ordinary modules, while Block, Residual, Repeat, and Parallel describe how they are wired. The resulting model can move unchanged from a teaching notebook to a complete pretraining run or a research ablation. OLM connects this readable model layer to tokenizers, local and streaming datasets, optimization, mixed precision, callbacks, checkpoints, and hardware-aware CPU, single-GPU, and single-node multi-GPU execution. We demonstrate the full path by tracing GPT-2 from diagram to code, launching a FineWeb-Edu training script, replacing one attention component, and letting AutoTrainer configure the available machine. The package includes 27 presets across nine familiar model families and documentation that progresses from LM fundamentals to architecture research. Validation shows close agreement with independent reference implementations, 90.6% four-GPU weak-scaling efficiency for a 348M-parameter workload, compact architecture edits, and positive early usability results. OLM is MIT-licensed and available through PyPI, GitHub, and its documentation site.

Comments: 8 pages, 3 figures, and 1 table. Website: this https URL. Code: this https URL

Subjects:

Computation and Language (cs.CL); Machine Learning (cs.LG); Software Engineering (cs.SE)

ACM classes: I.2.7

Cite as: arXiv:2607.16669 [cs.CL]

(or arXiv:2607.16669v1 [cs.CL] for this version)

https://doi.org/10.48550/arXiv.2607.16669

arXiv-issued DOI via DataCite (pending registration)

Submission history

From: Tavish Mankash [view email] [v1] Sat, 18 Jul 2026 06:55:18 UTC (411 KB)

Full-text links:

Access Paper:

View a PDF of the paper titled OpenLanguageModel: Readable and Composable Small-Language-Model Pretraining for Education and Research, by Tavish Mankash and 3 other authors

View PDF

HTML (experimental)

TeX Source

view license

Current browse context:

cs.CL

new | recent | 2026-07

Change to browse by:

cs cs.LG cs.SE

References & Citations

NASA ADS

Google Scholar

Semantic Scholar

Loading...

Data provided by:

Bibliographic Tools

Bibliographic and Citation Tools

Bibliographic Explorer Toggle

Bibliographic Explorer (What is the Explorer?)

Connected Papers Toggle

Connected Papers (What is Connected Papers?)

Litmaps Toggle

Litmaps (What is Litmaps?)

scite.ai Toggle

scite Smart Citations (What are Smart Citations?)

Code, Data, Media

Code, Data and Media Associated with this Article

alphaXiv Toggle

alphaXiv (What is alphaXiv?)

Links to Code Toggle

CatalyzeX Code Finder for Papers (What is CatalyzeX?)

DagsHub Toggle

DagsHub (What is DagsHub?)

GotitPub Toggle

Gotit.pub (What is GotitPub?)

Huggingface Toggle

Hugging Face (What is Huggingface?)

ScienceCast Toggle

ScienceCast (What is ScienceCast?)

Demos

Demos

Replicate Toggle

Replicate (What is Replicate?)

Spaces Toggle

Hugging Face Spaces (What is Spaces?)

Spaces Toggle

TXYZ.AI (What is TXYZ.AI?)

Related Papers

Recommenders and Search Tools

Link to Influence Flower

Influence Flower (What are Influence Flowers?)

Core recommender toggle

CORE Recommender (What is CORE?)

Author

Venue

Institution

Topic

About arXivLabs

arXivLabs: experimental projects with community collaborators

arXivLabs is a framework that allows collaborators to develop and share new arXiv features directly on our website.

Both individuals and organizations that work with arXivLabs have embraced and accepted our values of openness, community, excellence, and user data privacy. arXiv is committed to these values and only works with partners that adhere to them.

Have an idea for a project that will add value for arXiv's community? Learn more about arXivLabs.

Which authors of this paper are endorsers? | Disable MathJax (What is MathJax?)

Key points and analysis

Article intelligence

EngineersAdvanced

Key points

  • OLM provides readable model code that directly reflects architecture components for education and research.
  • It enables seamless movement from teaching notebooks to full pretraining runs with a complete pipeline.
  • The library includes 27 presets and validation shows 90.6% weak-scaling efficiency and positive usability.

Highlights and analysis are generated automatically and may contain errors. Check the original source.