Charl-AI / stochastic-caching
A tiny library for stochastic dataset caching in PyTorch.
☆40Updated last year
Alternatives and similar repositories for stochastic-caching:
Users that are interested in stochastic-caching are comparing it to the libraries listed below
- ☆50Updated 8 months ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆85Updated 3 years ago
- Easy Hypernetworks in Pytorch and Jax☆98Updated 2 years ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆86Updated last year
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆205Updated last year
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆126Updated 2 years ago
- ☆161Updated 3 months ago
- Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆115Updated 2 years ago
- FID computation in Jax/Flax.☆27Updated 7 months ago
- ☆73Updated 2 years ago
- Use Jax functions in Pytorch☆237Updated last year
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆127Updated last year
- Run PyTorch in JAX. 🤝☆225Updated 3 weeks ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 3 months ago
- Efficient optimizers☆180Updated this week
- Convert scikit-learn models to PyTorch modules☆161Updated 9 months ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆21Updated 4 months ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 years ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆49Updated 7 months ago
- ☆147Updated 2 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆147Updated 2 years ago
- ☆40Updated 3 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆168Updated 2 months ago
- ☆33Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆212Updated last week
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)☆89Updated 4 years ago