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
- Training HuggingFace models using fastai☆11Jul 22, 2021Updated 4 years ago
- ☆22Jan 23, 2024Updated 2 years ago
- Manifold-Mixup implementation for fastai V1☆19Oct 1, 2020Updated 5 years 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.
- 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☆50Jan 20, 2024Updated 2 years ago
- A case study of efficient training of large language models using commodity hardware.☆67Aug 4, 2022Updated 3 years ago
- Official implementation of 'A Large-Scale Exploration of mu-Transfer'☆32Jun 5, 2025Updated last year
- ☆19Nov 27, 2020Updated 5 years ago
- Testing various improvements to Ranger21 for 2022☆19Nov 6, 2024Updated last year
- Modular optimization library for PyTorch (work-in-progress).☆13Feb 4, 2026Updated 5 months ago
- ☆18Dec 2, 2024Updated last year
- Image scraper for DuckDuckGo and Google for creating DL datasets☆22Sep 18, 2020Updated 5 years ago
- 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 custom transfer agent for git-lfs that uses rclone to transfer files.☆13Jan 14, 2025Updated last year
- 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
- Mobile Viewer for W&B, built on top of Flutter.☆41Mar 2, 2024Updated 2 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)☆113Feb 10, 2026Updated 4 months ago
- Estimate derivatives with finite differences☆17Nov 18, 2024Updated last year
- ☆165Dec 13, 2023Updated 2 years ago
- Practical Deep Learning for Time Series / Sequential Data using fastai2/ Pytorch☆12Nov 12, 2020Updated 5 years ago
- 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.
- ☆10Apr 5, 2024Updated 2 years ago
- An recognition oriented deep learning framework for biometric sample quality assessment☆12Aug 24, 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 4 years ago
- TPU support for the fastai library☆14Apr 15, 2021Updated 5 years ago
- some mixture of experts architecture implementations☆27Mar 22, 2024Updated 2 years ago
- A react frontend app template/boilerplate, using Material-UI for UI.☆10May 13, 2024Updated 2 years ago
- ☆45Oct 13, 2023Updated 2 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 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.
- Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing☆15Jul 2, 2022Updated 4 years ago
- ☆62Mar 4, 2022Updated 4 years ago
- ☆15Jan 5, 2026Updated 6 months ago
- Code you can use jointly with fastai☆93Nov 30, 2020Updated 5 years ago
- RevBiFPN: The Fully Reversible Bidirectional Feature Pyramid Network☆15Oct 18, 2022Updated 3 years ago
- Brax + Pufferlib + CARBS for gpu-accelerated robotics RL☆12Jun 12, 2025Updated last year
- A collection of inference modules for fastai2☆91Oct 6, 2022Updated 3 years ago