Fix a bug in setting up custom kernels

lanl / hippynn

python library for atomistic machine learning

https://lanl.github.io/hippynn/

Other

59 stars 22 forks source link

Closed tautomer closed 1 year ago

tautomer commented 1 year ago

Now "auto" will use pytorch when numba or cupy is not installed. Plus some formatting with black (now this file should match the global formatting).