juvi21 / self-compressing-nn-jax Goto Github PK
View Code? Open in Web Editor NEWImplementation of Self-Compressing Neural Networks in JAX https://arxiv.org/pdf/2301.13142 ported from tinygrad https://github.com/geohot/ai-notebooks/blob/master/mnist_self_compression.ipynb
License: MIT License