ChanBong / s4-jaxLinks
Following research on S4 in jax
☆14Updated 3 years ago
Alternatives and similar repositories for s4-jax
Users that are interested in s4-jax are comparing it to the libraries listed below
Sorting:
- This is a question bank for practicing Machine Learning for Interviews.☆33Updated 2 years ago
- ☆16Updated last year
- Reproduction of the CVPR'21 paper Distilling Knowledge via Knowledge Review for the ML Reproducibility Challenge 2021☆10Updated 3 years ago
- graftr: an interactive shell to view and edit PyTorch checkpoints.☆113Updated 4 years ago
- Python library for argument and configuration management☆54Updated 2 years ago
- Simple python template☆41Updated last year
- gpu tester detects broken and slow gpus in a cluster☆70Updated 2 years ago
- Context Manager to profile the forward and backward times of PyTorch's nn.Module☆83Updated last year
- Examples of using PyTorch hooks, as covered in my YouTube tutorial video.☆33Updated last year
- GAN models implemented with Pytorch Lightning and Hydra configuration☆34Updated 3 years ago
- An open source implementation of CLIP.☆32Updated 2 years ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- Cyclemoid implementation for PyTorch☆90Updated 3 years ago
- FID computation in Jax/Flax.☆27Updated 11 months ago
- Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.☆22Updated 2 years ago
- Contains code for the paper "Vision Transformers are Robust Learners" (AAAI 2022).☆125Updated 2 years ago
- This project shows how to derive the total number of training tokens from a large text dataset from 🤗 datasets with Apache Beam and Data…☆27Updated 2 years ago
- Automatically take good care of your preemptible TPUs☆36Updated 2 years ago
- A minimalistic boiler plate code for training pytorch models☆13Updated last year
- Optimized library for large-scale extraction of frames and audio from video.☆204Updated last year
- ☆74Updated 2 years ago
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆41Updated 4 months ago
- Developing adversarial examples and showing their semantic generalization for the OpenAI CLIP model (https://github.com/openai/CLIP)☆26Updated 4 years ago
- Implementation of Discrete Key / Value Bottleneck, in Pytorch☆88Updated last year
- A small demonstration of using WebDataset with ImageNet and PyTorch Lightning☆74Updated last year
- ☆34Updated 9 months ago
- JAX implementation ViT-VQGAN☆83Updated 2 years ago
- ☆10Updated last year
- PixMix: Dreamlike Pictures Comprehensively Improve Safety Measures (CVPR 2022)☆108Updated 2 years ago
- All about the fundamentals and working of Diffusion Models☆158Updated 2 years ago