more-itertools
optree
torch
