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:
- Train vision models using JAX and π€ transformersβ100Updated 2 weeks ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offloadβ131Updated 3 years ago
- Optimized library for large-scale extraction of frames and audio from video.β204Updated 2 years ago
- FID computation in Jax/Flax.β29Updated last year
- 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
- Automatically take good care of your preemptible TPUsβ37Updated 2 years ago
- Contrastive Language-Image Pretrainingβ144Updated 3 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorchβ252Updated 3 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAIβ97Updated 3 years ago
- β314Updated last year
- Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of newβ¦β125Updated last year
- Automatic gradient descentβ215Updated 2 years ago
- JAX implementation ViT-VQGANβ62Updated 3 years ago
- A JAX implementation of the continuous time formulation of Consistency Modelsβ85Updated 2 years ago
- Unofficial JAX implementations of deep learning research papersβ160Updated 3 years ago
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)β189Updated 3 years ago
- Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arenaβ206Updated 2 years ago
- β52Updated last year
- An alternative to convolution in neural networksβ258Updated last year
- JAX implementation ViT-VQGANβ82Updated 3 years ago
- Various handy scripts to quickly setup new Linux and Windows sandboxes, containers and WSL.β40Updated this week
- β75Updated 3 years ago
- A case study of efficient training of large language models using commodity hardware.β68Updated 3 years ago
- Named tensors with first-class dimensions for PyTorchβ332Updated 2 years ago
- gpu tester detects broken and slow gpus in a clusterβ72Updated 2 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorchβ88Updated 2 years ago
- Amos optimizer with JEstimator lib.β82Updated last year
- Implementation of π» Mirasol, SOTA Multimodal Autoregressive model out of Google Deepmind, in Pytorchβ91Updated 2 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library β€οΈβ57Updated 2 years ago
- Exploring an idea where one forgets about efficiency and carries out attention across each edge of the nodes (tokens)β55Updated 9 months ago