Aleph-Alpha-Research / scalingView external linksLinks
Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for training large language models.
☆66Nov 18, 2025Updated 2 months ago
Alternatives and similar repositories for scaling
Users that are interested in scaling are comparing it to the libraries listed below
Sorting:
- Code repository for the public reproduction of the language modelling experiments on "MatFormer: Nested Transformer for Elastic Inference…☆31Nov 14, 2023Updated 2 years ago
- Manage ML configuration with pydantic☆16Jan 25, 2026Updated 2 weeks ago
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- ☆47Jan 18, 2024Updated 2 years ago
- A framework for PyTorch to enable fault management for collective communication libraries (CCL) such as NCCL☆20Updated this week
- Code for the paper "Function-Space Learning Rates"☆25Jun 3, 2025Updated 8 months ago
- ☆21Mar 3, 2025Updated 11 months ago
- Scratchpad/Chain-of-Thought Prompts☆12Jun 6, 2022Updated 3 years ago
- ☆10Dec 21, 2024Updated last year
- [ICLR 26] The official code repository for the paper "Mirage or Method? How Model–Task Alignment Induces Divergent RL Conclusions".☆15Updated this week
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 8 months ago
- ☆12Jan 4, 2024Updated 2 years ago
- A blog where I write about research papers and blog posts I read.☆12Nov 20, 2024Updated last year
- A statistical framework for graph anomaly detection.☆17Sep 23, 2018Updated 7 years ago
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32May 25, 2024Updated last year
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- [WIP] Transformer to embed Danbooru labelsets☆13Mar 31, 2024Updated last year
- nanoGPT using Equinox☆15Mar 3, 2023Updated 2 years ago
- ☆34Sep 10, 2024Updated last year
- A library for unit scaling in PyTorch☆133Jul 11, 2025Updated 7 months ago
- Asynchronous P2P communication backend for decentralized pipeline parallelism☆41Jun 9, 2025Updated 8 months ago
- ☆33Nov 4, 2024Updated last year
- Graph model execution API for Candle☆17Jul 27, 2025Updated 6 months ago
- A spoken version of the textual story cloze benchmark☆20Aug 6, 2023Updated 2 years ago
- A Proxy service using FastAPI and Protocol Buffers (Proto3)☆13Jun 17, 2023Updated 2 years ago
- DeMo: Decoupled Momentum Optimization☆198Dec 2, 2024Updated last year
- OpenDiLoCo: An Open-Source Framework for Globally Distributed Low-Communication Training☆562Jan 13, 2025Updated last year
- Writing FLUX in Triton☆41Sep 22, 2024Updated last year
- Debug print operator for cudagraph debugging☆14Aug 2, 2024Updated last year
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆24Jan 9, 2026Updated last month
- BenchBench is a Python package to evaluate multi-task benchmarks.☆18Oct 12, 2025Updated 4 months ago
- Codebase for adaptive continual memory☆13Aug 15, 2023Updated 2 years ago
- NanoGPT-speedrunning for the poor T4 enjoyers☆73Apr 22, 2025Updated 9 months ago
- WIP☆93Aug 13, 2024Updated last year
- ☆250Dec 2, 2024Updated last year
- Triton kernels for Flux☆22Jul 7, 2025Updated 7 months ago
- GGML bindings that aim to be idiomatic Rust rather than directly corresponding to the C/C++ interface☆19Sep 25, 2023Updated 2 years ago