OpTree: Optimized PyTree Utilities - View it on GitHub