numpy

[all]
torch
torchvision
torch-scatter
jax[cuda12]
matplotlib

[examples]
matplotlib

[jax]
jax[cuda12]

[jax-cpu]
jax

[torch]
torch
torch-scatter
torchvision
