A collection of optimizers, some arcane others well known, for Flax.
☆29Aug 6, 2021Updated 4 years ago
Alternatives and similar repositories for flaxOptimizers
Users that are interested in flaxOptimizers are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Your fruity companion for transformers☆14May 25, 2022Updated 4 years ago
- Jax implementation of the AdaHessian optimizer☆19Mar 11, 2021Updated 5 years ago
- ☆12Mar 16, 2022Updated 4 years ago
- Training HuggingFace models using fastai☆11Jul 22, 2021Updated 4 years ago
- Common Voice Generator using Speech Synthesizer☆14Jul 28, 2021Updated 4 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Manifold-Mixup implementation for fastai V1☆19Oct 1, 2020Updated 5 years ago
- ☆16May 14, 2025Updated last year
- Trace LLM calls (and others) and visualize them in WandB, as interactive SVG or using a streaming local webapp☆14Feb 18, 2025Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Jan 20, 2024Updated 2 years ago
- To be a next-generation DL-based phenotype prediction from genome mutations.☆19May 17, 2021Updated 5 years ago
- Collection of the latest, greatest, deep learning optimizers (for Pytorch) - CNN, NLP suitable☆218Apr 4, 2021Updated 5 years ago
- ☆27Apr 23, 2026Updated last month
- A case study of efficient training of large language models using commodity hardware.☆67Aug 4, 2022Updated 3 years ago
- PyTorch Code for the Paper: "Exploiting Uncertainty of Loss Landscape for Stochastic Optimization [Bhaskara et al. (2019)]☆16Apr 30, 2026Updated last month
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- ☆19Nov 27, 2020Updated 5 years ago
- Modular optimization library for PyTorch (work-in-progress).☆13Feb 4, 2026Updated 4 months ago
- ☆14Aug 28, 2019Updated 6 years ago
- Code for "Accelerating Training with Neuron Interaction and Nowcasting Networks" [ICLR 2025]☆29Feb 20, 2026Updated 3 months ago
- Train a bidirectional or normal LSTM recurrent neural network to generate text on a free GPU using any dataset. Just upload your text fil…☆12Jan 29, 2019Updated 7 years ago
- implementation of replicated softmax model☆13Sep 25, 2014Updated 11 years ago
- A rio-tiler plugin to create tiles using TMS (WebMercator or others)☆11Oct 13, 2020Updated 5 years ago
- Ranger deep learning optimizer rewrite to use newest components☆341Mar 17, 2026Updated 3 months ago
- Qwen3-0.6B megakernel: 527 tok/s decode on RTX 3090 (3.8x faster than PyTorch)☆110Feb 10, 2026Updated 4 months 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.
- Estimate derivatives with finite differences☆17Nov 18, 2024Updated last year
- ☆166Dec 13, 2023Updated 2 years ago
- Implementation of Kronecker Attention in Pytorch☆20Sep 12, 2020Updated 5 years ago
- Age classification from text using PAN16, blogs, Fisher Callhome, and Cancer Forum☆18Jul 1, 2022Updated 3 years ago
- ☆45Oct 13, 2023Updated 2 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 5 years ago
- Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing☆15Jul 2, 2022Updated 3 years ago
- An implementation of a Brownian motion using ClojureScript with re-frame and Highcharts☆11Feb 8, 2019Updated 7 years ago
- Code you can use jointly with fastai☆93Nov 30, 2020Updated 5 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- RevBiFPN: The Fully Reversible Bidirectional Feature Pyramid Network☆15Oct 18, 2022Updated 3 years ago
- Code for OGB competition☆12Feb 24, 2024Updated 2 years ago
- Brax + Pufferlib + CARBS for gpu-accelerated robotics RL☆12Jun 12, 2025Updated last year
- Testing of Neural Topic Modeling for Japanese articles☆13Jul 24, 2019Updated 6 years ago
- A Citation Manager and Zotero Integration for RemNote! Cite research all within your knowledge base!☆29Jan 22, 2026Updated 4 months ago
- A simple API to launch Python functions to run on multiple ranked processes, mpify is designed to enable interactive multiprocessing expe…☆11Sep 29, 2020Updated 5 years ago
- Starlight: A Kernel Optimizer for GPU Processing☆16Jan 10, 2024Updated 2 years ago