4 projects
openmmjax-cu13
OpenMM plugin for exported JAX/XLA models
openmmjax-cu12
OpenMM plugin for exported JAX/XLA models
e3nn-jax
Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translations, and mirrors.
e3nn
Equivariant convolutional neural networks for the group E(3) of 3 dimensional rotations, translations, and mirrors.