For object inspection in Jupyter if doing more numpy type things, check out penzai: https://github.com/google-deepmind/penzai