BobMcDear / flaimLinks
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
☆23Updated 7 months ago
Alternatives and similar repositories for flaim
Users that are interested in flaim are comparing it to the libraries listed below
Sorting:
- ☆32Updated 2 years ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆29Updated 5 years ago
- AdaCat☆49Updated 3 years ago
- Recursive Leasting Squares (RLS) with Neural Network for fast learning☆59Updated 2 years ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆55Updated 10 months ago
- High-performance tokenized language data-loader for Python C++ extension☆14Updated last year
- ☆29Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆32Updated 2 years ago
- 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
- RWKV model implementation☆38Updated 2 years ago
- recipe for training fully-featured self supervised image jepa models☆12Updated 7 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆61Updated 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 Metaformer, but in an autoregressive manner☆26Updated 3 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated 2 years ago
- Implementation of a holodeck, written in Pytorch☆18Updated 2 years ago
- PyTorch reimplementation of the paper "HyperMixer: An MLP-based Green AI Alternative to Transformers" [arXiv 2022].☆18Updated 3 years ago
- A dashboard for exploring timm learning rate schedulers☆19Updated last year
- A JAX nn library☆22Updated 4 months ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- ☆31Updated last week
- A stateful pytree library for training neural networks.☆22Updated 5 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆92Updated last year
- Implementation of a Light Recurrent Unit in Pytorch☆49Updated last year
- ☆35Updated last year
- ☆24Updated last year
- Official code for Long Expressive Memory (ICLR 2022, Spotlight)☆71Updated 3 years ago
- Implementation of Spectral State Space Models☆16Updated last year