typing-extensions

[default]
optree

[jax]
jax>=0.4.7

[numpy]
numpy
optree

[torch]
torch
optree
