SymTorch allows you to approximate the behaviour of components within deep learning models with symbolic equations using PySR.

SymTorch is available on PyPI as torch-symbolic:

pip install torch-symbolic

Our paper is available on arXiv. You can also visit the accompanying website for more information.

Full documentation is available at ReadTheDocs.

MIT License