BobMcDear / flaimLinks
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
☆23Updated 8 months ago
Alternatives and similar repositories for flaim
Users that are interested in flaim are comparing it to the libraries listed below
Sorting:
- A dashboard for exploring timm learning rate schedulers☆19Updated last year
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆29Updated 5 years ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆32Updated 2 years ago
- Implementation of a holodeck, written in Pytorch☆18Updated 2 years ago
- AdaCat☆49Updated 3 years ago
- High-performance tokenized language data-loader for Python C++ extension☆14Updated last year
- RWKV model implementation☆37Updated 2 years ago
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- PyTorch reimplementation of the paper "HyperMixer: An MLP-based Green AI Alternative to Transformers" [arXiv 2022].☆18Updated 3 years ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated last year
- Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"☆59Updated 2 years ago
- Local Attention - Flax module for Jax☆22Updated 4 years ago
- A Python library for Continual Inference Networks in PyTorch☆54Updated 10 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- A GPT, made only of MLPs, in Jax☆59Updated 4 years ago
- Implementation of Metaformer, but in an autoregressive manner☆26Updated 3 years ago
- A stateful pytree library for training neural networks.☆22Updated 5 months ago
- Visualising Losses in Deep Neural Networks☆16Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆61Updated 3 years ago
- Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing☆49Updated 4 years ago
- ☆32Updated 2 years ago
- Utilities for PyTorch distributed☆25Updated 11 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆55Updated 10 months ago
- ☆21Updated 2 years ago
- ☆31Updated 2 weeks ago
- Recursive Leasting Squares (RLS) with Neural Network for fast learning☆59Updated 2 years ago
- Implementation of the Remixer Block from the Remixer paper, in Pytorch☆36Updated 4 years ago
- Implementation of a Light Recurrent Unit in Pytorch☆49Updated last year
- Implementation of Spectral State Space Models☆16Updated last year