Skip to content

Your favorite basis functions for equivariant models in one place. Code borrowed from e3x

License

Notifications You must be signed in to change notification settings

atomicarchitects/BasisLib

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

29 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

BasisLib

Copy of Numpy-based basis functions utilities from e3x.

Useful external APIs:

  • BasisLib.so3.generate_clebsch_gordan_lookup_table: Current .npz cache supports upto lmax=30
  • BasisLib.so3.generate_spherical_harmonics_lookup_table: Current .npz cache supports upto lmax=15

Refer to BasisLib.so3.spherical_harmonics and BasisLib.so3.clesch_gordan for their usage.

Workflow for generating more coefficients

python -m BasisLib.so3.clebsch_gordan_lut --path $HOME/BasisLib/BasisLib/so3_clebsch_gordan_lut.npz --max_degree 50 --num_processes 47

About

Your favorite basis functions for equivariant models in one place. Code borrowed from e3x

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages