Alternatives and similar repositories for stochastic-caching

Users that are interested in stochastic-caching are comparing it to the libraries listed below

• lucidrains / logavgexp-torch
  Implementation of LogAvgExp for Pytorch
  ☆36Updated 2 months ago
• andravin / spio
  Memory-Efficient CUDA kernels for training ConvNets with PyTorch.
  ☆41Updated 4 months ago
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆97Updated last year
• maxxxzdn / implicit-steerable-kernels
  Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]
  ☆28Updated 4 months ago
• lucidrains / nystrom-attention
  Implementation of Nyström Self-attention, from the paper Nyströmformer
  ☆135Updated 3 months ago
• balbasty / torch-interpol
  High-order spline interpolation in PyTorch
  ☆67Updated 9 months ago
• CSCfi / pytorch-ddp-examples
  ☆50Updated last year
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆242Updated last year
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆177Updated 2 weeks ago
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆108Updated 7 months ago
• kazukiosawa / asdl
  ASDL: Automatic Second-order Differentiation Library for PyTorch
  ☆188Updated 6 months ago
• facebookresearch / FFCV-SSL
  FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.
  ☆208Updated last year
• hmdolatabadi / denoising_diffusion
  Unofficial Implementation of "Denoising Diffusion Probabilistic Models" in PyTorch(Lightning)
  ☆91Updated 4 years ago
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆64Updated 3 weeks ago
• gregorbachmann / scaling_mlps
  ☆51Updated last year
• nikhilvyas / SOAP
  ☆190Updated 6 months ago
• NKI-AI / ece_loss
  A PyTorch implementation of the Exclusive Cross Entropy Loss.
  ☆20Updated 2 years ago
• kvfrans / splus
  ☆104Updated 2 weeks ago
• Lightning-Universe / paper-AAVAE
  ☆47Updated 2 years ago
• lucidrains / Adan-pytorch
  Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch
  ☆252Updated 2 years ago
• jiaweizzhao / ZerO-initialization
  ☆74Updated 2 years ago
• HomebrewML / revlib
  Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
  ☆127Updated 2 years ago
• YannDubs / Invariant-Self-Supervised-Learning
  Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"
  ☆41Updated 2 years ago
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆91Updated 3 years ago
• lucidrains / x-unet
  Implementation of a U-net complete with efficient attention as well as the latest research findings
  ☆282Updated last year
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆37Updated last month
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆46Updated 2 years ago
• ivan-chai / torch-linear-assignment
  Batch computation of the linear assignment problem on GPU.
  ☆83Updated 5 months ago
• facebookresearch / torchdim
  Named tensors with first-class dimensions for PyTorch
  ☆331Updated 2 years ago
• Artur-Galstyan / statedict2pytree
  ☆43Updated 3 weeks ago