-
Hello,
Firstly, I would like to extend my gratitude for the impressive work you have presented in your recent paper and the accompanying repository.
I watched the accompanying video from your I…
-
Dear all,
great project! I look forward to using this in Jax, especially since the Jax.scipy implementation does not work well.
I am trying to pip-install it, but I got the following error
```bas…
-
Eq.h5 files with pressure saved as a PowerSeriesProfile crash when running eq.solve()
Attaching error report. This issue can be circumvented by switching to SplineProfile. Dario has also seen it an…
-
It would be great to see support for jax and jaxlib - https://github.com/google/jax
Jax actually installs fine, but jaxlib requires ml_dtypes (https://github.com/jax-ml/ml_dtypes) which is not comp…
-
## Describe the bug
As far as I understand it, I should be able to use this library paired with jax via your backend switching (depending on the input types)? However, I am getting a `jax.errors.Trac…
act65 updated
1 month ago
-
Following the most recent commit, the jax.tree module is now being used instead of jax.tree_*. This is change requires jax >= 0.4.25. However, the are still many parts of the repository that are still…
-
### Description
```python
import jax
import jax.numpy as jnp
def f(x):
def cond(carry):
i, x, acc = carry
return i < x.shape[0]
def body(carry):
i, x, acc = carry
r…
-
when i run pip install -r requirements.txt happened that
`Collecting jax==0.3.23 (from -r requirements.txt (line 34))
Downloading http://mirrors.aliyun.com/pypi/packages/43/ce/7ce437f917b8bd0e55db…
-
Hi authors,
The codebase is very helpful! And thanks for your efforts in putting everything together.
While I was trying the model, I saw that the model weights of this pytorch version were port…
-
Ever since originally adopting [`autograd`](https://github.com/HIPS/autograd), we've been concerned that most of the development energy from [`autograd`](https://github.com/HIPS/autograd) has moved to…