Minimal pretraining script for language modeling in PyTorch. Supporting torch compilation and DDP. It includes a model implementation and a data preprocessing script.
☆50Apr 27, 2026Updated 5 months ago
Alternatives and similar repositories for plainLM
Users that are interested in plainLM are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- A CLI to run reproducible evaluation across EuroHPC clusters☆21Updated this week
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 7 months ago
- Real-Time RTUs☆13Mar 20, 2026Updated 6 months ago
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆66Mar 11, 2025Updated last year
- Pretraining codebase for Apertus models, based on Megatron-LM☆23Oct 2, 2026Updated last week
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- Source code for "Neural Anisotropy Directions"☆16Nov 17, 2020Updated 5 years ago
- Tools for managing BibTeX bibliographies: automatically update preprints to published versions and filter to only cited references.☆116Sep 26, 2026Updated 2 weeks ago
- This repository is for setting-up cuda-9/8, nvidia-396/387/384 driver, OpenCV-3.3, ROS Kinetic, Tensorflow-1.11/1.7/1.4/1.2.1, Pytorch-0.…☆30Jul 7, 2022Updated 4 years ago
- The first collection of surrogate benchmarks for Joint Architecture and Hyperparameter Search.☆15Mar 22, 2023Updated 3 years ago
- Official code for "Algorithmic Capabilities of Random Transformers" (NeurIPS 2024)☆16Sep 28, 2024Updated 2 years ago
- Linear Attention for Efficient Bidirectional Sequence Modeling☆18May 13, 2025Updated last year
- Agar.io for Continual Reinforcement Learning☆28Aug 27, 2026Updated last month
- ☆70Aug 28, 2026Updated last month
- Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.☆26Feb 20, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Irene is a python package that aims to be a toolkit for global optimization problems that can be realized algebraically. It generalizes L…☆15Updated this week
- ☆17Oct 25, 2022Updated 3 years ago
- Optimization methods course at MIPT☆11Apr 24, 2018Updated 8 years ago
- Hessian trace estimation using PyTorch and Hutch++☆20Oct 29, 2020Updated 5 years ago
- ☆22Mar 25, 2024Updated 2 years ago
- This repository regroups learning ressources about performance estimation problems☆15Mar 18, 2026Updated 6 months ago
- Benchmarking Optimizers for LLM Pretraining☆61May 3, 2026Updated 5 months ago
- 📄 Evidence Retrieval and Claim Verification for the FEVER shared task using Transformer Networks☆12Feb 21, 2020Updated 6 years ago
- Implementation of the psquare algorithm for quantile value estimation☆11Apr 21, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆16Jan 16, 2023Updated 3 years ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆12Jul 19, 2024Updated 2 years ago
- Randomized Linear Algebra in Python☆13Mar 21, 2017Updated 9 years ago
- A Signal Propagation Perspective for Pruning Neural Networks at Initialization☆14Jun 23, 2020Updated 6 years ago
- Code for implementing central flows☆56Sep 5, 2025Updated last year
- flexible meta-learning in jax☆16Oct 19, 2023Updated 2 years ago
- ☆33Jul 18, 2026Updated 2 months ago
- JAX implementation of the Mistral 7b v0.1 model☆13Mar 27, 2024Updated 2 years ago
- Things I care about☆13Jul 10, 2022Updated 4 years ago
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- MoMo: Momentum Models for Adaptive Learning Rates☆19Jun 12, 2024Updated 2 years ago
- A set of solutions to ETHZ ROS lectures☆13Jul 19, 2017Updated 9 years ago
- ☆22May 24, 2020Updated 6 years ago
- ☆37Updated this week
- Parameter-Free Optimizers for Pytorch☆132Apr 23, 2024Updated 2 years ago
- The ros_qml package☆12Sep 7, 2015Updated 11 years ago
- Reimplementation of ToMNet with some extensions for RL as well☆14Apr 28, 2018Updated 8 years ago