Dian Wu

Initialize independent and dependent caches separately in ARNN

5

As the PR for RNN is merged, now I can start upstreaming the changes in my branch for MPS-RNN into the master branch, and I'd like to split them into...

Weight normalization

2

Is there an equivalent to `flax.optim.WeightNorm`? As `flax.optim` is effectively deprecated in favor of optax, I would like to see it implemented in optax.

Looking from the Python API, `SpinBasis` supports `symmetries`, but `SpinfulFermionBasis` and `SpinlessFermionBasis` do not. Is there any plan to support them?

Hi @twesterhout , I'd say thank you again for this package! Recently we had a [paper](https://arxiv.org/abs/2407.05917) using this package and we're planning to publish the code. It would make things...