If you are classically trained I highly recommend Mark Levine’s book[1]. It digs into the theory of jazz, alongside improvisation. It does assume a working knowledge of basic theory, so not so great for those starting from scratch.

[1]: https://www.amazon.com/Jazz-Piano-Book-Mark-Levine/dp/096147...