optree

Optimized PyTree Utilities

homepage: https://optree.readthedocs.io/en/latest/

Available installations

optree version	Supported CPU targets	Supported GPU targets	EESSI version	Module
0.14.1	generic: aarch64, x86_64 Arm: a64fx, neoverse_n1, neoverse_v1, nvidia/grace AMD: zen2, zen3, zen4 Intel: haswell, skylake_avx512, sapphirerapids, icelake, cascadelake	(none)	2025.06	optree/0.14.1-GCCcore-13.3.0