Schedule free optimiser implemented in JAX using Optimistix
☆14May 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:
- Dynamical Systems with JAX☆12Jun 3, 2026Updated 4 months ago
- A Wadler–Lindig pretty printer for Python☆49Updated this week
- Code for the paper "Function-Space Learning Rates"☆23Jun 3, 2025Updated last year
- Tensor Parallelism with JAX + Shard Map☆11Sep 29, 2023Updated 3 years ago
- manipulating cointegrated pairs to achieve a market-neutral strategy that outperforms indices☆11Jan 12, 2021Updated 5 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.
- A snappy + easy + pretty TUI debugger for Python.☆70May 22, 2026Updated 5 months ago
- Prototype for sparse automatic differentiation in JAX + Julia☆24Jun 17, 2025Updated last year
- JAX translation of boltz☆32Aug 11, 2026Updated 2 months ago
- Gaussian Markov Random Fields (GMRFs) and Integrated Nested Laplace Approximation (INLA)☆24Apr 17, 2024Updated 2 years ago
- A haptic authoring tool☆12May 19, 2025Updated last year
- ESM2 protein language models in JAX/Flax☆18Oct 10, 2022Updated 4 years ago
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 years ago
- Automate Finding and Fetching Academic Citations☆11Oct 10, 2024Updated 2 years ago
- Your favourite classical machine learning algos on the GPU/TPU☆23Dec 14, 2025Updated 10 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- ☆11Oct 19, 2026Updated last week
- ☆72Sep 11, 2026Updated last month
- A complete example of batched refactorization in cuSOLVER.☆12Dec 22, 2021Updated 4 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated last month
- A project to illustrate the power of Jax for simulating astrochemical kinetics.☆11Oct 12, 2026Updated 2 weeks ago
- Python library for SBGN☆12Oct 22, 2026Updated last week
- ☆11Mar 29, 2025Updated last year
- Simple visualized cetz editor☆12Oct 8, 2026Updated 3 weeks ago
- Code for the paper: "Generalized Discrete Diffusion from Snapshots"☆14Apr 2, 2026Updated 6 months ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- coloring terminal text with intensities (used for plotting probability, entropy with tokens)☆12Oct 11, 2024Updated 2 years ago
- A functional training loops library for JAX☆87Feb 13, 2024Updated 2 years ago
- [NeurIPS 2026] Official implementation of Contrastive Distribution Matching for Amortized Sequential Monte Carlo in Discrete Diffusion☆15Oct 1, 2026Updated 3 weeks ago
- ☆10Jul 27, 2026Updated 3 months ago
- A JAX implementation of stochastic addition.☆13Aug 15, 2022Updated 4 years ago
- ☆13Nov 14, 2024Updated last year
- Parallel hyperparameter tuning with JAX☆40Sep 30, 2026Updated 3 weeks ago
- ☆12Oct 7, 2025Updated last year
- Code and data for the paper ”Counting the Bugs in ChatGPT's Wugs: A Multilingual Investigation into the Morphological Capabilities of a L…☆10Dec 3, 2023Updated 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.
- A few models converted from caffe to CoreMLs format.☆15Jun 6, 2017Updated 9 years ago
- Multiple dispatch over abstract array types in JAX.☆143Oct 21, 2026Updated last week
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆17Jan 7, 2025Updated last year
- Scratchpad/Chain-of-Thought Prompts☆12Jun 6, 2022Updated 4 years ago
- Official Code Repo for Paper: Posterior Refinement☆14Sep 22, 2026Updated last month
- ☆14Mar 2, 2025Updated last year
- Open-sourcing the code associated with the NeurIPS 2025 paper "Efficient Transition Matrices to Enable State Tracking in State-Space Mode…☆16Dec 8, 2025Updated 10 months ago