Charl-AI / stochastic-caching
A tiny library for stochastic dataset caching in PyTorch.
☆39Updated 9 months ago
Related projects ⓘ
Alternatives and complementary repositories for stochastic-caching
- Implementation of the Adan (ADAptive Nesterov momentum algorithm) Optimizer in Pytorch☆247Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆124Updated 2 years ago
- ☆22Updated last month
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆77Updated 10 months ago
- FFCV-SSL Fast Forward Computer Vision for Self-Supervised Learning.☆203Updated last year
- ☆73Updated 2 years ago
- Named tensors with first-class dimensions for PyTorch☆322Updated last year
- ASDL: Automatic Second-order Differentiation Library for PyTorch☆179Updated 3 months ago
- Easy Hypernetworks in Pytorch and Jax☆96Updated last year
- Replicating and dissecting the git-re-basin project in one-click-replication Colabs☆36Updated 2 years ago
- ☆47Updated 2 years ago
- Load tensorboard event logs as pandas DataFrames for scientific plotting; Supports both PyTorch and TensorFlow☆181Updated 3 months ago
- Implementation of a U-net complete with efficient attention as well as the latest research findings☆267Updated 6 months ago
- ☆41Updated 5 months ago
- Implementation of LogAvgExp for Pytorch☆32Updated 2 years ago
- Multidimensional indexing for tensors☆113Updated last year
- Efficient CUDA kernels for training convolutional neural networks with PyTorch.☆35Updated this week
- ☆164Updated last year
- TF/Keras code for DiffStride, a pooling layer with learnable strides.☆124Updated 2 years ago
- Cyclemoid implementation for PyTorch☆87Updated 2 years ago
- Efficient optimizers☆80Updated this week
- The spiritual successor to knockknock for PyTorch Lightning, get notified when your training ends☆76Updated this week
- Use Jax functions in Pytorch☆228Updated last year
- A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to fac…☆216Updated this week
- ☆35Updated 9 months ago
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).☆104Updated 2 years ago
- Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …☆128Updated last month
- Convert scikit-learn models to PyTorch modules☆157Updated 6 months ago
- ☆129Updated last week
- Simple Python scripts to clean up and flatten ArXiv LaTeX submissions.☆63Updated 2 years ago