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
- An open source implementation of CLIP.☆32Updated 2 years ago
- Semantic Segmentation with Pytorch-Lightning☆63Updated 4 years ago
- Texture mapping with variational auto-encoders☆40Updated 3 years ago
- AdaCat☆49Updated 2 years ago
- Minimal implementation of adaptive gradient clipping (https://arxiv.org/abs/2102.06171) in TensorFlow 2.☆84Updated 3 years ago
- Cyclemoid implementation for PyTorch☆89Updated 3 years ago
- ☆73Updated 2 years ago
- A simple implementation of a deep linear Pytorch module☆19Updated 4 years ago
- GAN models implemented with Pytorch Lightning and Hydra configuration☆34Updated 2 years ago
- Generate bird's-eye views of conference proceedings.☆24Updated 4 months ago
- Large dataset storage format for Pytorch☆45Updated 3 years ago
- JAX implementation of Learning to learn by gradient descent by gradient descent☆27Updated 6 months ago
- ☆15Updated 3 years ago
- Simple tooling for marking deprecated functions or classes and re-routing to the new successors' instance.☆51Updated this week
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 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
- A selection of neural network models ported from torchvision for JAX & Flax.☆44Updated 4 years ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆108Updated 2 years ago
- ☆47Updated 2 years ago
- Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️☆54Updated 2 years ago
- ☆26Updated 2 years ago
- An active learning library for Pytorch based on Lightning-Fabric.☆79Updated 11 months ago
- Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"☆18Updated last year
- ☆67Updated last year
- Efficiently Composable Data Augmentation on the GPU with Jax☆33Updated 8 months ago
- A GPT, made only of MLPs, in Jax☆57Updated 3 years ago
- Another attempt at a long-context / efficient transformer by me☆37Updated 3 years ago
- 🎲 Iterable dataset resampling in PyTorch☆91Updated 3 years ago
- This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.☆14Updated 3 years ago
- Code for the paper PermuteFormer☆42Updated 3 years ago