PyPI page
Home page
Author:
Austin Glover, Vivek Bharadwaj, Aydin Buluc, James Demmel
Summary:
A fast GPU JIT kernel generator for the Clebsch-Gordon Tensor Product
Latest version:
0.6.6
Required dependencies:
jinja2
|
ninja
|
numpy
|
setuptools
Optional dependencies:
cmake
|
cuequivariance
|
cuequivariance-ops-torch-cu12
|
cuequivariance-torch
|
e3nn
|
matplotlib
|
nanobind
|
pre-commit
|
pytest
|
pytest-check
|
pytest-subtests
|
ruff
|
scikit-build-core
|
setuptools-scm
|
torch_geometric
|
tqdm
Downloads last day:
42,189
Downloads last week:
210,249
Downloads last month:
343,322