Charl-AI / stochastic-caching
A tiny library for stochastic dataset caching in PyTorch.
☆40Updated 11 months ago
Alternatives and similar repositories for stochastic-caching:
Users that are interested in stochastic-caching are comparing it to the libraries listed below
- ☆47Updated 7 months ago
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆251Updated 2 years ago
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- ☆47Updated 2 years ago
- Use Jax functions in Pytorch☆232Updated last year
- Easy Hypernetworks in Pytorch and Jax☆96Updated 2 years ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆203Updated last year
- Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆84Updated 3 years ago
- Named tensors with first-class dimensions for PyTorch☆321Updated last year
- ☆72Updated 2 years ago
- Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆101Updated 2 months ago
- Implementation of Nyström Self-attention, from the paper Nyströmformer☆123Updated last year
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆125Updated 2 years ago
- Simple Python scripts to clean up and flatten ArXiv LaTeX submissions.☆64Updated 2 years ago
- Running Jax in PyTorch Lightning☆86Updated last month
- Code of "Deep invariant networks with differentiable augmentation layers"☆18Updated 2 years ago
- ☆163Updated 2 years ago
- Train ImageNet *fast* in 500 lines of code with FFCV☆136Updated 8 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆106Updated 2 years ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆188Updated 5 months ago
- Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆209Updated 9 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆84Updated last year
- FID computation in Jax/Flax.☆26Updated 6 months ago
- A simple program to calculate and visualize the FLOPs and Parameters of Pytorch models, with handy CLI and easy-to-use Python API.☆124Updated 2 months ago
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆182Updated last month
- Unofficial JAX implementations of deep learning research papers☆153Updated 2 years ago
- Code for the paper: Rotating Features for Object Discovery☆49Updated 5 months ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆21Updated 2 months ago
- Transformers with doubly stochastic attention☆44Updated 2 years ago