Schedule free optimiser implemented in JAX using Optimistix
☆15May 29, 2024Updated 2 years ago
Alternatives and similar repositories for schedule_free_optx
Users that are interested in schedule_free_optx are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- Tidy autoregressive inference in JAX☆15Sep 1, 2025Updated 9 months ago
- Code for the paper "Function-Space Learning Rates"☆23Jun 3, 2025Updated last year
- JAX translation of boltz☆28Aug 4, 2025Updated 10 months ago
- A Wadler–Lindig pretty printer for Python☆48Apr 20, 2026Updated 2 months ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- A small library of paramaterizations and parameter constraints for PyTrees.☆62Jan 23, 2026Updated 5 months ago
- A snappy + easy + pretty TUI debugger for Python.☆70May 22, 2026Updated last month
- Coordinates in JAX☆35May 20, 2026Updated last month
- Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)☆22Apr 17, 2024Updated 2 years ago
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- It's not ROS... but it is ROS-y! Simple, fast inter-process message passing for distributed Python processes☆11May 29, 2026Updated last month
- ☆19Dec 4, 2025Updated 6 months ago
- ☆13Jun 3, 2024Updated 2 years ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆21Jul 20, 2024Updated last year
- A haptic authoring tool☆12May 19, 2025Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆16Jan 7, 2025Updated last year
- Python library for SBGN☆11May 19, 2026Updated last month
- Your favourite classical machine learning algos on the GPU/TPU☆23Dec 14, 2025Updated 6 months ago
- Numerical quadrature with JAX☆84Jun 22, 2026Updated last week
- Code for "Log Neural Controlled Differential Equations" (ICML 2024) and "Structured Linear CDEs" (NeurIPS 2025, Spotlight)☆33May 5, 2026Updated last month
- ☆15Dec 20, 2024Updated last year
- Multiple dispatch over abstract array types in JAX.☆142Jun 22, 2026Updated last week
- 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.
- A functional training loops library for JAX☆88Feb 13, 2024Updated 2 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆47Jun 15, 2026Updated 2 weeks ago
- ☆72Nov 15, 2024Updated last year
- A complete example of batched refactorization in cuSOLVER.☆13Dec 22, 2021Updated 4 years ago
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated last year
- Simple visualized cetz editor☆12Jun 22, 2026Updated last week
- ☆18Aug 24, 2024Updated last year
- Some deep learning models written with mxnet and C++11.☆12Feb 6, 2018Updated 8 years ago
- A web based URDF visualizer using three.js☆22Nov 15, 2022Updated 3 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Sequence-level 1F1B schedule for LLMs.☆19Jun 4, 2024Updated 2 years ago
- BONNI: Bayesian Optimization via Neural Network surrogates and Interior Point Optimization☆34Jun 19, 2026Updated last week
- A MkDocs plugin for embedding interactive jupyter notebooks in your docs via jupyterlite.☆25Dec 18, 2025Updated 6 months ago
- Curated list of JAX Resources and Packages☆45Jun 7, 2026Updated 3 weeks ago
- ☆63Oct 3, 2024Updated last year
- Differentiable collision detection for capsules in JAX.☆56Dec 2, 2023Updated 2 years ago
- Experiments Notebook of "Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanism"☆16Apr 30, 2025Updated last year