Charl-AI / stochastic-cachingLinks
A tiny library for stochastic dataset caching in PyTorch.
☆44Updated 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:
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆110Updated 2 months ago
 - Implementation of Nyström Self-attention, from the paper Nyströmformer☆141Updated 7 months ago
 - Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI☆96Updated 3 years ago
 - Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆252Updated 3 years ago
 - Implementation of a U-net complete with efficient attention as well as the latest research findings☆288Updated last year
 - Implementation of LogAvgExp for Pytorch☆37Updated 6 months ago
 - Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch☆132Updated 2 weeks ago
 - Code repository for the ICLR 2022 paper "FlexConv: Continuous Kernel Convolutions With Differentiable Kernel Sizes" https://openreview.ne…☆116Updated 2 years ago
 - ☆75Updated 2 years ago
 - Use Jax functions in Pytorch☆253Updated 2 years ago
 - Running Jax in PyTorch Lightning☆113Updated 10 months ago
 - Convert scikit-learn models to PyTorch modules☆166Updated last year
 - ASDL: Automatic Second-order Differentiation Library for PyTorch☆190Updated 10 months ago
 - Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆131Updated 3 years ago
 - A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆242Updated last month
 - Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆188Updated 2 weeks ago
 - ☆59Updated last year
 - Integral Neural Networks in PyTorch☆127Updated 11 months ago
 - A parallel ODE solver for PyTorch☆270Updated last year
 - Named tensors with first-class dimensions for PyTorch☆331Updated 2 years ago
 - TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 3 years ago
 - ☆164Updated 2 years ago
 - TensorLy-Torch: Deep Tensor Learning with TensorLy and PyTorch☆81Updated last year
 - Drop-in replacement for any ResNet with a significantly reduced memory footprint and better representation capabilities☆208Updated last year
 - Transformers with doubly stochastic attention☆49Updated 3 years ago
 - A repository to house some personal attempts to beat some state-of-the-art for medical datasets☆100Updated last year
 - Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)☆91Updated 5 years ago
 - Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆205Updated last year
 - [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated 2 years ago
 - FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆210Updated 2 years ago