/jax_pruning

A JAX implementation of structure and unstructure pruning

Primary LanguagePythonMIT LicenseMIT

Stargazers