Adding comment on O(3) backend in README
Browse files
README.md
CHANGED
|
@@ -20,6 +20,9 @@ We provide weights for mimicking the temporal evolution of three different syste
|
|
| 20 |
- **quartz**: Weights for crystalline quartz using a time interval of 30 fs.
|
| 21 |
- **water**: Weights for liquid bulk water using a time interval of 5 fs.
|
| 22 |
|
|
|
|
|
|
|
|
|
|
| 23 |
Below we provide an overview of our architecture. For more information we refer to our [preprint](https://arxiv.org/) and [code](https://github.ibm.com/AD-TrajCast/trajcast).
|
| 24 |
|
| 25 |
<p align="center">
|
|
|
|
| 20 |
- **quartz**: Weights for crystalline quartz using a time interval of 30 fs.
|
| 21 |
- **water**: Weights for liquid bulk water using a time interval of 5 fs.
|
| 22 |
|
| 23 |
+
> **_Note_:**
|
| 24 |
+
> We provide each model based on two different O(3) backends: [e3nn](https://github.com/e3nn/e3nn) and [cuEquivariance](https://docs.nvidia.com/cuda/cuequivariance/). Choose the state dictionary and config.yaml dependent on whether you have CUDA and cuEquivariance installed. Please note that depending on the device used to initialize a model with the cuEquivariance backend, some parameter names may differ.
|
| 25 |
+
|
| 26 |
Below we provide an overview of our architecture. For more information we refer to our [preprint](https://arxiv.org/) and [code](https://github.ibm.com/AD-TrajCast/trajcast).
|
| 27 |
|
| 28 |
<p align="center">
|