pengyan510 / torcheck
☆38Updated 2 years ago
Alternatives and similar repositories for torcheck:
Users that are interested in torcheck are comparing it to the libraries listed below
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 4 months ago
- ☆73Updated 2 years ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- AdaCat☆49Updated 2 years ago
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆53Updated 2 years ago
- Implementation of "compositional attention" from MILA, a multi-head attention variant that is reframed as a two-step attention process wi…☆50Updated 2 years ago
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- Implementation of numerous Vision Transformers in Google's JAX and Flax.☆22Updated 2 years ago
- ☆47Updated 2 years ago
- Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation☆12Updated 3 years ago
- a lightweight transformer library for PyTorch☆71Updated 3 years ago
- A PyTorch Lightning extension that accelerates and enhances foundation model experimentation with flexible fine-tuning schedules.☆62Updated last month
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Collection of snippets for PyTorch users☆25Updated 2 years ago
- ☆15Updated 3 years ago
- notebooks of cool EBM visualizations☆15Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆51Updated 2 months ago
- FID computation in Jax/Flax.☆27Updated 7 months ago
- ☆47Updated 4 years ago
- A collection of Models, Datasets, DataModules, Callbacks, Metrics, Losses and Loggers to better integrate pytorch-lightning with transfor…☆47Updated last year
- ☆68Updated last year
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆84Updated 3 years ago
- Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"☆18Updated last year
- Semantic Segmentation with Pytorch-Lightning☆63Updated 4 years ago
- ☆14Updated 2 years ago
- A Python library for Continual Inference Networks in PyTorch☆49Updated 2 weeks ago