-
Have you thought about using ```jax.numpy``` instead of ```numpy```? I am using ```py3nj``` for my research and now I am porting the code to JAX. Do you think that the way the code is written would be…
-
The pasted code below demonstrates that mode-coupling matrices can be computed much more efficiently than it is currently done in `pspy`. The demo pasted below runs roughly ten times faster with the `…
-
Hi,
I was looking online for fast python implementations of the 3j symbols. My use of the 3j symbols comes from computations in spherical harmonics space, and as such, I'm computing the symbols for…
-
I'm trying to run a variant of the mace model with hidden irreps of order l=4 (`20x0e + 20x1o + 20x2e + 20x3o + 20x4e`) and I get an error on the symmetric contraction:
```python
prod = Equiva…