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

@misc{symtorch2026,
  title        = {SymTorch: A Framework for Symbolic Distillation of Deep Neural Networks},
  author       = {Tan, Elizabeth S.Z. and Soubki, Adil and Cranmer, Miles},
  year         = {2026},
  eprint       = {2602.21307},
  archivePrefix= {arXiv},
  primaryClass = {cs.LG},
  url          = {https://arxiv.org/abs/2602.21307}
}