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 4 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:
- Discretax is a light weight collection of state space models implemented in JAX ⚡️☆38Mar 9, 2026Updated 6 months ago
- Code for the paper: Why Transformers Need Adam: A Hessian Perspective☆65Mar 11, 2025Updated last year
- ICLR Blog Track 2025☆21Sep 21, 2025Updated 11 months ago
- Pretraining codebase for Apertus models, based on Megatron-LM☆22Sep 25, 2025Updated 11 months ago
- Source code for "Neural Anisotropy Directions"☆16Nov 17, 2020Updated 5 years ago
- 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.
- 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
- Linear Attention for Efficient Bidirectional Sequence Modeling☆18May 13, 2025Updated last year
- Gradient-based Hyperparameter Optimization Over Long Horizons☆14Sep 29, 2021Updated 4 years ago
- Agar.io for Continual Reinforcement Learning☆28Aug 27, 2026Updated 3 weeks ago
- ☆70Aug 28, 2026Updated 3 weeks ago
- Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.☆26Feb 20, 2025Updated last year
- [ICLR 2023] "Sparsity May Cry: Let Us Fail (Current) Sparse Neural Networks Together!" Shiwei Liu, Tianlong Chen, Zhenyu Zhang, Xuxi Chen…☆28Aug 29, 2023Updated 3 years ago
- Lattice Recurrent Unit: Improving Convergence and Statistical Efficiency for Sequence Modeling☆36Jan 30, 2018Updated 8 years ago
- ☆17Oct 25, 2022Updated 3 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Code for Paper (Policy Optimization in RLHF: The Impact of Out-of-preference Data)☆29Dec 19, 2023Updated 2 years ago
- ☆16Oct 21, 2025Updated 10 months ago
- Optimization methods course at MIPT☆11Apr 24, 2018Updated 8 years ago
- Official repository for the paper "Exploring the Promise and Limits of Real-Time Recurrent Learning" (ICLR 2024)☆13Jun 11, 2025Updated last year
- ☆13Mar 10, 2026Updated 6 months ago
- Official implementation for the paper "Controlled Sparsity via Constrained Optimization"☆12Aug 10, 2022Updated 4 years ago
- Implementation of the psquare algorithm for quantile value estimation☆11Apr 21, 2024Updated 2 years ago
- [CVPR 2022] "Quarantine: Sparsity Can Uncover the Trojan Attack Trigger for Free" by Tianlong Chen*, Zhenyu Zhang*, Yihua Zhang*, Shiyu C…☆28Oct 5, 2022Updated 3 years ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆12Jul 19, 2024Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- [ICML2023] Instant Soup Cheap Pruning Ensembles in A Single Pass Can Draw Lottery Tickets from Large Models. Ajay Jaiswal, Shiwei Liu, Ti…☆11Nov 28, 2023Updated 2 years ago
- Things I care about☆13Jul 10, 2022Updated 4 years ago
- MoMo: Momentum Models for Adaptive Learning Rates☆19Jun 12, 2024Updated 2 years ago
- Quantized Evolution Strategies: High-precision Fine-tuning of Quantized LLMs at Low-precision Cost☆21May 14, 2026Updated 4 months ago
- ☆22May 24, 2020Updated 6 years ago
- ☆36Aug 14, 2026Updated last month
- An implementation of LassoNet for arbitrary network architectures☆14Nov 4, 2022Updated 3 years ago
- PyTorch implementation of the paper The Lottery Ticket Hypothesis for Object Recognition☆23Apr 22, 2021Updated 5 years ago
- Parameter-Free Optimizers for Pytorch☆132Apr 23, 2024Updated 2 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- Reimplementation of ToMNet with some extensions for RL as well☆14Apr 28, 2018Updated 8 years ago
- Bayesian methods in deep learning Summer School☆19Aug 31, 2017Updated 9 years ago
- Convert CVXPY expressions to PyTorch expressions☆18Jul 8, 2025Updated last year
- Implementation of the fast weight product key memory from Sakana AI☆20Updated this week
- Bring ROS to any Linux distributions.☆15Jun 12, 2019Updated 7 years ago
- Code to create bugged python scripts for OpenAssistant Training, maintained by https://twitter.com/Cyndesama☆24Jul 23, 2023Updated 3 years ago
- Neural model of hierarchical reinforcement learning☆16Sep 14, 2017Updated 9 years ago