Charl-AI / stochastic-cachingLinks
A tiny library for stochastic dataset caching in PyTorch.
☆42Updated last year
Alternatives and similar repositories for stochastic-caching
Users that are interested in stochastic-caching are comparing it to the libraries listed below
Sorting:
- Implementation of LogAvgExp for Pytorch☆36Updated 2 months ago
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆41Updated 4 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆97Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆28Updated 4 months ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆135Updated 3 months ago
- High-order spline interpolation in PyTorch☆67Updated 9 months ago
- ☆50Updated last year
- Use Jax functions in Pytorch☆242Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆177Updated 2 weeks ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆108Updated 7 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆188Updated 6 months ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆208Updated last year
- Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)☆91Updated 4 years ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆64Updated 3 weeks ago
- ☆51Updated last year
- ☆190Updated 6 months ago
- A PyTorch implementation of the Exclusive Cross Entropy Loss.☆20Updated 2 years ago
- ☆104Updated 2 weeks ago
- ☆47Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Updated 2 years ago
- ☆74Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆91Updated 3 years ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆282Updated last year
- Efficiently Composable Data Augmentation on the GPU with Jax☆37Updated last month
- Transformers with doubly stochastic attention☆46Updated 2 years ago
- Batch computation of the linear assignment problem on GPU.☆83Updated 5 months ago
- Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
- ☆43Updated 3 weeks ago