google-deepmind / compressed_visionLinks
β130Updated 2 years ago
Alternatives and similar repositories for compressed_vision
Users that are interested in compressed_vision are comparing it to the libraries listed below
Sorting:
- Optimized library for large-scale extraction of frames and audio from video.β204Updated last year
- Train vision models using JAX and π€ transformersβ98Updated this week
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offloadβ129Updated 3 years ago
- Internet Explorer explores the web in a self-supervised manner to progressively find relevant examples that improve performance on a desiβ¦β163Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorchβ251Updated 2 years ago
- Contrastive Language-Image Pretrainingβ144Updated 2 years ago
- β307Updated last year
- Automatic gradient descentβ210Updated 2 years ago
- FID computation in Jax/Flax.β28Updated last year
- Automatically take good care of your preemptible TPUsβ36Updated 2 years ago
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of newβ¦β124Updated last year
- Unofficial JAX implementations of deep learning research papersβ156Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)β187Updated 3 years ago
- JAX implementation ViT-VQGANβ59Updated 3 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arenaβ205Updated 2 years ago
- JAX implementation ViT-VQGANβ83Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAIβ94Updated 3 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library β€οΈβ55Updated 2 years ago
- A JAX implementation of the continuous time formulation of Consistency Modelsβ85Updated 2 years ago
- CLOOB training (JAX) and inference (JAX and PyTorch)β72Updated 3 years ago
- β75Updated 2 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorchβ88Updated 2 years ago
- Implementation of fused cosine similarity attention in the same style as Flash Attentionβ215Updated 2 years ago
- gpu tester detects broken and slow gpus in a clusterβ70Updated 2 years ago
- β185Updated last year
- Easy Hypernetworks in Pytorch and Jaxβ104Updated 2 years ago
- Named tensors with first-class dimensions for PyTorchβ331Updated 2 years ago
- Texture mapping with variational auto-encodersβ40Updated 3 years ago
- β91Updated 2 years ago
- Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'β38Updated 3 years ago