Author here: I specifically mentioned the `egg` crate because of it's capabilities for symbolic math. You can check a minimal application of arithmetic differentiation and some simplification rules in their repo here:

https://github.com/egraphs-good/egg/blob/main/tests/math.rs