BobMcDear / flaimLinks
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
☆20Updated 2 weeks 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 11 months ago
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 3 years ago
- Implementation of Spectral State Space Models☆16Updated last year
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆29Updated 4 years ago
- ☆17Updated 10 months ago
- ☆18Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- Training hybrid models for dummies.☆23Updated 5 months ago
- Minimum Description Length probing for neural network representations☆18Updated 4 months ago
- Shows how to do parameter ensembling using differential evolution.☆10Updated 3 years ago
- RWKV model implementation☆38Updated last year
- ☆27Updated last year
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 5 years ago
- Residual Quantization Autoencoder, used for interpreting LLMs☆12Updated 5 months ago
- Repo for solving arc problems with an Neural Cellular Automata☆16Updated last month
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆25Updated 5 months ago
- An attempt to merge ESBN with Transformers, to endow Transformers with the ability to emergently bind symbols☆16Updated 3 years ago
- Personal solutions to the Triton Puzzles☆19Updated 11 months ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- Implementation of a holodeck, written in Pytorch☆18Updated last year
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated last year
- GoldFinch and other hybrid transformer components☆10Updated last month
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆30Updated 2 weeks ago
- See https://github.com/cuda-mode/triton-index/ instead!☆11Updated last year
- LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence☆60Updated 3 years ago
- ☆11Updated last year
- ☆16Updated last year
- My explorations into editing the knowledge and memories of an attention network☆35Updated 2 years ago
- ☆31Updated last week