BobMcDear / flaim
Flax Image Models - State-of-the-art pre-trained vision backbones for Flax.
☆19Updated last year
Alternatives and similar repositories for flaim:
Users that are interested in flaim are comparing it to the libraries listed below
- Training hybrid models for dummies.☆20Updated 3 months ago
- Implementation of Spectral State Space Models☆16Updated last year
- High-performance tokenized language data-loader for Python C++ extension☆13Updated 8 months ago
- Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"☆28Updated 4 years ago
- ☆18Updated last year
- Shows how to do parameter ensembling using differential evolution.☆10Updated 3 years ago
- A stateful pytree library for training neural networks.☆22Updated 2 years ago
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆31Updated last year
- ☆13Updated 7 months ago
- Simplifying parsing of large jsonline files in NLP Workflows☆12Updated 3 years ago
- Minimum Description Length probing for neural network representations☆19Updated 2 months ago
- ☆11Updated 11 months ago
- Jax like function transformation engine but micro, microjax☆30Updated 5 months ago
- This is a port of Mistral-7B model in JAX☆32Updated 9 months ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- Personal solutions to the Triton Puzzles☆18Updated 9 months ago
- A dashboard for exploring timm learning rate schedulers☆19Updated 4 months ago
- Implements EvoNorms B0 and S0 as proposed in Evolving Normalization-Activation Layers.☆11Updated 4 years ago
- RWKV model implementation☆37Updated last year
- Describe the format of image/text datasets☆11Updated 2 years ago
- Easily serialize dataclasses to and from tensors (PyTorch, NumPy)☆18Updated 4 years ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)☆32Updated 10 months ago
- ☆27Updated last year
- Implementation of a holodeck, written in Pytorch☆17Updated last year
- Local Attention - Flax module for Jax☆20Updated 3 years ago
- ☆39Updated 2 years ago
- ☆20Updated 3 years ago
- ☆15Updated 2 years ago
- ☆17Updated 7 months ago
- See https://github.com/cuda-mode/triton-index/ instead!☆11Updated 11 months ago