-
The conversion interface to gpu4pyscf repo is mostly completed. I'm considering to create a similar conversion function for pyscfad. What is your opinion?
sunqm updated
8 months ago
-
### Description
[JAX](https://github.com/google/jax) includes a numpy compatible `jax.numpy` module which has a bunch of nice features (automatic differentiation, jit compilation, vectorized mappin…
-
Hi!
Our team is currently developing sparse symmetric and non-symmetric solvers in JAX (implicitly restarted arnoldi and lanzcos). I have it from hearsay that there are efforts to support this nati…
-
Vmapping across `y0` with any method using `AbstractGaussNewton` throws a TypeError. MWE
```python
import jax
import jax.numpy as jnp
import optimistix as optx
def rosenbrock(x, args):
…
-
### System information
flax Version: 0.9.0
jax Version: 0.4.34
jaxlib Version: 0.4.34
on my laptop, running in CPU only mode.
flax Version: 0.8.5
jax Version: 0.4.33
jaxlib Version: 0.4.33
on goo…
-
### System information
- OS Platform and Distribution (e.g., Linux Ubuntu 16.04): `Ubuntu 22.04.4 LTS x86_64`
- Flax, jax, jaxlib versions:
```
Name: flax
Version: 0.8.4
---
Name: jax
Version…
-
Hi everyone,
I'm asking because I noticed a very weird behaviour in jax-unirep that, even with a good amount of effort trying to narrow down the cause, I cannot seem to find it.
The weird behavi…
-
##### Description
I have a (mandatory) 'body' parameter in an API call that's not reliably deserialized. If I use 'Postman' to construct a test query the whole query comes in a single packet a…
-
| Q | A
| --------------------------------- | -------
| Bug or feature request? | Feature Request
| Which Swagger-Core version? | 1.5.17…
-
TPU type: v3-8
[TPU software version](https://cloud.google.com/tpu/docs/supported-tpu-versions#tpu_software_versions): tpu-vm-tf-2.9.1
Installed PIP packages / `pip freeze` output (click to ex…