I believe you are talking about this book.

Dive into Deep Learning: https://d2l.ai/