BobMcDear / flaimLinks
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
☆23Updated 3 months ago
Alternatives and similar repositories for flaim
Users that are interested in flaim are comparing it to the libraries listed below
Sorting:
- High-performance tokenized language data-loader for Python C++ extension☆13Updated last year
- Implementation of a holodeck, written in Pytorch☆18Updated last year
- A stateful pytree library for training neural networks.☆22Updated 2 weeks ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated 2 years ago
- ☆32Updated last year
- A dashboard for exploring timm learning rate schedulers☆19Updated 9 months ago
- ☆13Updated 3 months ago
- AdaCat☆49Updated 3 years ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆29Updated 5 years ago
- Implementation of a Transformer using ReLA (Rectified Linear Attention) from https://arxiv.org/abs/2104.07012☆49Updated 3 years ago
- Visualising Losses in Deep Neural Networks☆16Updated last year
- ☆31Updated 2 months ago
- PyTorch reimplementation of the paper "HyperMixer: An MLP-based Green AI Alternative to Transformers" [arXiv 2022].☆17Updated 3 years ago
- Local Attention - Flax module for Jax☆22Updated 4 years ago
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆59Updated 3 years ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)☆53Updated 5 months ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- A JAX nn library☆21Updated last week
- ☆21Updated 2 years ago
- Utilities for PyTorch distributed☆25Updated 6 months ago
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- Recursive Leasting Squares (RLS) with Neural Network for fast learning☆56Updated last year
- A python library for highly configurable transformers - easing model architecture search and experimentation.☆49Updated 3 years ago
- Implementation of a Light Recurrent Unit in Pytorch☆48Updated 11 months ago
- A GPT, made only of MLPs, in Jax☆58Updated 4 years ago
- High performance pytorch modules☆18Updated 2 years ago
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- RWKV model implementation☆38Updated 2 years ago
- Fine-Tuning Pre-trained Transformers into Decaying Fast Weights☆19Updated 2 years ago
- Utilities for Training Very Large Models☆58Updated 11 months ago