Pytrees + static analysis ❤️ - View it on GitHub