Charl-AI / stochastic-caching
A tiny library for stochastic dataset caching in PyTorch.
☆43Updated last year
Alternatives and similar repositories for stochastic-caching:
Users that are interested in stochastic-caching are comparing it to the libraries listed below
- ☆22Updated 5 months ago
- Use Jax functions in Pytorch☆239Updated last year
- Implementation of LogAvgExp for Pytorch☆34Updated 3 years ago
- ☆50Updated 9 months ago
- ☆73Updated 2 years ago
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆86Updated 3 years ago
- Memory-Efficient CUDA kernels for training ConvNets with PyTorch.☆40Updated last month
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆131Updated 2 weeks ago
- The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch☆58Updated last month
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆185Updated 4 months ago
- ☆47Updated 2 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆207Updated last year
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Run PyTorch in JAX. 🤝☆233Updated last month
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆90Updated last year
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 11 months ago
- 🚀 A powerful library for efficient training of Neural Fields at scale.☆28Updated last year
- Easy Hypernetworks in Pytorch and Jax☆100Updated 2 years ago
- Cyclemoid implementation for PyTorch☆89Updated 3 years ago
- Convert scikit-learn models to PyTorch modules☆162Updated 10 months ago
- Named tensors with first-class dimensions for PyTorch☆319Updated last year
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆127Updated 2 years ago
- ☆36Updated last year
- Flow-matching algorithms in JAX☆86Updated 7 months ago
- ☆42Updated 2 weeks ago
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆49Updated 8 months ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆103Updated 4 months ago
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated 2 years ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago