Schedule free optimiser implemented in JAX using Optimistix
☆15May 29, 2024Updated last year
Alternatives and similar repositories for schedule_free_optx
Users that are interested in schedule_free_optx are comparing it to the libraries listed below
Sorting:
- Code for the paper "Function-Space Learning Rates"☆25Jun 3, 2025Updated 8 months ago
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 6 months ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- Prototype for sparse automatic differentiation in JAX + Julia☆20Jun 17, 2025Updated 8 months ago
- ☆13Jun 3, 2024Updated last year
- nanoGPT using Equinox☆15Mar 3, 2023Updated 2 years ago
- Sequence-level 1F1B schedule for LLMs.☆19Jun 4, 2024Updated last year
- ☆18Aug 24, 2024Updated last year
- A small library of paramaterizations and parameter constraints for PyTrees.☆50Jan 23, 2026Updated last month
- A Wadler–Lindig pretty printer for Python☆44Jan 16, 2026Updated last month
- JAX translation of boltz☆23Aug 4, 2025Updated 6 months ago
- Coordinates in JAX☆29Jan 31, 2026Updated last month
- Code for "Log Neural Controlled Differential Equations" (ICML 2024) and "Structured Linear CDEs" (NeurIPS 2025, Spotlight)☆30Jan 26, 2026Updated last month
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 years ago
- Supporting code for the blog post on modular manifolds.☆117Sep 26, 2025Updated 5 months ago
- ☆24Oct 21, 2024Updated last year
- Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)☆22Apr 17, 2024Updated last year
- ☆25Jun 4, 2024Updated last year
- Your favourite classical machine learning algos on the GPU/TPU☆21Dec 14, 2025Updated 2 months ago
- ☆63Oct 3, 2024Updated last year
- ☆24Jun 4, 2024Updated last year
- Scaling is a distributed training library and installable dependency designed to scale up neural networks, with a dedicated module for tr…☆66Nov 18, 2025Updated 3 months ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆20Jul 20, 2024Updated last year
- Algorithm for blind estimation of reverberation time☆35Jun 6, 2024Updated last year
- [ICLR 2025] SDTT: a simple and effective distillation method for discrete diffusion models☆47Updated this week
- ☆70Nov 15, 2024Updated last year
- Mask-Enhanced Autoregressive Prediction: Pay Less Attention to Learn More☆34May 17, 2025Updated 9 months ago
- Numerical quadrature with JAX☆77Feb 16, 2026Updated last week
- An unofficial implementation of the Infini-gram model proposed by Liu et al. (2024)☆33Jun 19, 2024Updated last year
- manipulating cointegrated pairs to achieve a market-neutral strategy that outperforms indices☆12Jan 12, 2021Updated 5 years ago
- [ICML 2024] Official Repository for the paper "Transformers Get Stable: An End-to-End Signal Propagation Theory for Language Models"☆10Jul 19, 2024Updated last year
- A python algorithm to change the pitch of the voice in real time☆13Dec 13, 2020Updated 5 years ago
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- ☆55Updated this week
- A snappy + easy + pretty TUI debugger for Python.☆69Oct 3, 2025Updated 4 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆46Dec 9, 2024Updated last year
- Make Git monorepos like a boss.☆13Dec 18, 2025Updated 2 months ago
- Asynchronous Wiki Engine☆10Sep 4, 2023Updated 2 years ago
- A fine-mapping method integrating GWAS summary statistics and functional annotation data☆11Dec 28, 2023Updated 2 years ago