diff --git a/README.md b/README.md index c91f618..bd40e58 100644 --- a/README.md +++ b/README.md @@ -1,4 +1,6 @@ # Sparsity-preserving gradient utility tools for PyTorch +![License](https://img.shields.io/github/license/cai4cai/torchsparsegradutils) ![Code Style: Black](https://img.shields.io/badge/code%20style-black-000000.svg) + A collection of utility functions to work with PyTorch sparse tensors. This is work-in-progress, here be dragons. Currenly available features with backprop include: