google-deepmind / hierarchical_perceiverLinks
☆27Updated 2 weeks ago
Alternatives and similar repositories for hierarchical_perceiver
Users that are interested in hierarchical_perceiver are comparing it to the libraries listed below
Sorting:
- Official PyTorch Implementation of the Longhorn Deep State Space Model☆50Updated 6 months ago
- FID computation in Jax/Flax.☆27Updated 10 months ago
- ☆51Updated 11 months ago
- ☆29Updated 6 months ago
- TPU pod commander is a package for managing and launching jobs on Google Cloud TPU pods.☆20Updated 11 months ago
- ☆53Updated last year
- ☆32Updated last year
- PyTorch Package For Quasimetric Learning☆42Updated 7 months ago
- Minimal but scalable implementation of large language models in JAX☆34Updated 7 months ago
- Exploration into the proposed "Self Reasoning Tokens" by Felipe Bonetto☆56Updated last year
- ☆46Updated last year
- GPT implementation in Flax☆18Updated 3 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆100Updated 2 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆29Updated 11 months ago
- ☆51Updated last year
- ☆53Updated 8 months ago
- Automatically take good care of your preemptible TPUs☆36Updated 2 years ago
- Train vision models using JAX and 🤗 transformers☆97Updated last month
- ☆33Updated 8 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆81Updated last year
- Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper☆81Updated 3 years ago
- ☆19Updated 2 weeks ago
- ☆33Updated 2 years ago
- Code for the paper "Function-Space Learning Rates"☆20Updated last month
- A simple library for scaling up JAX programs☆137Updated 7 months ago
- Code Release for "Broken Neural Scaling Laws" (BNSL) paper☆58Updated last year
- simple bibtex generator for any text with \cite{}☆31Updated 10 months ago
- flexible meta-learning in jax☆14Updated last year
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆53Updated 2 years ago
- Blog post☆17Updated last year