Core API
========

Sparsification
--------------

.. autofunction:: compresso.topk_ste

.. autoclass:: compresso.TopKSparsify
   :members:

Sparse Representations
----------------------

.. autoclass:: compresso.SRPTensor
   :members:

Sparse Autoencoders
-------------------

.. autoclass:: compresso.TopKSAE
   :members:

.. autoclass:: compresso.TopKSAEConfig
   :members:

.. autoclass:: compresso.TopKSAETrainer
   :members:

.. autoclass:: compresso.L1Normalize
   :members:

.. autoclass:: compresso.L2Normalize
   :members:

Sparse Parameters
-----------------

.. autoclass:: compresso.MaskedParam
   :members:

.. autoclass:: compresso.SRPParam
   :members:
   
Utilities
---------

.. autoclass:: compresso.SparsityController
   :members:

.. autofunction:: compresso.exponential_decay

.. autofunction:: compresso.save_srp_tensor

.. autofunction:: compresso.load_srp_tensor