A collection of optimizers, some arcane others well known, for Flax.
☆29Aug 6, 2021Updated 5 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
- JAX + Haiku package for the discrete VAE used for DALL·E.☆11Apr 10, 2021Updated 5 years ago
- ☆12Mar 16, 2022Updated 4 years ago
- EfficientNet implementation in PyTorch - https://arxiv.org/abs/1905.11946☆10Jul 30, 2019Updated 7 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Training HuggingFace models using fastai☆11Jul 22, 2021Updated 5 years ago
- ☆11Oct 11, 2022Updated 4 years ago
- ☆17May 14, 2025Updated last year
- ☆16Apr 29, 2020Updated 6 years ago
- 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
- To be a next-generation DL-based phenotype prediction from genome mutations.☆19May 17, 2021Updated 5 years ago
- A case study of efficient training of large language models using commodity hardware.☆67Aug 4, 2022Updated 4 years ago
- Code for the paper: "Invertible CNN-Based Super Resolution with Downsampling Awareness" by Andrew Geiss and Joseph C. Hardin, Nov 2020☆12Nov 11, 2020Updated 5 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Official implementation of 'A Large-Scale Exploration of mu-Transfer' (CoRR 2024)☆32Jun 5, 2025Updated last year
- ☆14Aug 28, 2019Updated 7 years ago
- Ranger deep learning optimizer rewrite to use newest components☆339Mar 17, 2026Updated 7 months ago
- ☆10May 21, 2017Updated 9 years ago
- 🤓 A tutorial on the Discretized Logistic Mixture Likelihood (DLML)☆12Aug 10, 2024Updated 2 years ago
- 🍱 fsspec interface for Weights & Biases (wandb)☆10Oct 27, 2022Updated 4 years ago
- An recognition oriented deep learning framework for biometric sample quality assessment☆12Aug 24, 2023Updated 3 years ago
- A rio-tiler plugin to create tiles using TMS (WebMercator or others)☆11Oct 13, 2020Updated 6 years ago
- implementation of replicated softmax model☆13Sep 25, 2014Updated 12 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- A Python library for pruning and visualizing Keras Neural Networks' structure and weights☆11Dec 24, 2019Updated 6 years ago
- Converting EfficientNet to Pytorch for use with fastai☆27Jun 5, 2019Updated 7 years ago
- Stick Breaking Variational Autoencoder with HDP prior☆10Nov 29, 2017Updated 8 years ago
- ☆46Oct 13, 2023Updated 3 years ago
- ☆62Mar 4, 2022Updated 4 years ago
- SoftEther VPN (ALLFREE Edition Repository): Open Cross-platform Multi-protocol VPN Software. This branch is remove function restrictions …☆10Dec 1, 2022Updated 3 years ago
- ☆30Jan 17, 2022Updated 4 years ago
- Deep learning lectures I am holding for the MSc on Data Science and Scientific Computing☆16Jul 2, 2022Updated 4 years ago
- A curated list of resources for using fast.ai's nbdev☆19Jun 24, 2020Updated 6 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 collection of inference modules for fastai2☆90Oct 6, 2022Updated 4 years ago
- ☆36Jan 23, 2024Updated 2 years ago
- Starlight: A Kernel Optimizer for GPU Processing☆17Jan 10, 2024Updated 2 years ago
- Building the Bi-LSTM & the CNN-GAN models to compose Classical Music in different eras☆12Aug 2, 2021Updated 5 years ago
- A Citation Manager and Zotero Integration for RemNote! Cite research all within your knowledge base!☆29Jan 22, 2026Updated 9 months ago
- This repository has moved, please visit https://github.com/ai2cm/pace for the latest development of fv3core.☆13Dec 21, 2022Updated 3 years ago
- MTPNet: Multi-Grained Target Perception for Unified Activity Cliff Prediction☆14May 19, 2025Updated last year