-
### 🚀 The feature, motivation and pitch
It would be great to have a general parallel prefix sum (associative scan) operation in PyTorch, something like [associative_scan](https://jax.readthedocs.io…
-
### Feature description
Currently PyBOP supports inference of some equivalent circuit model parameters, where those parameters remain constant across the depth of discharge. Cell kinetics and resista…
-
One thing that would be very helpful for implementors is providing clarity on how JAX-RS Resource classes and Providers are instantiated when used with CDI. Presently, there's some significant behavio…
-
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…
-
### Description
Hello, there,
I observe large time difference using jit and ask for help.
```python
key = jax.random.PRNGKey(666)
data = jax.random.normal(key, shape=[1000])
data = jax.devic…
-
Hi @patrick-kidger,
Apologises for opening an issue as this is probably something I've missed, but is it at all possible to separate an `eqx.Module` object into a pure function and its parameters? …
-
Looking at the list of dependencies [jax](https://github.com/google/jax) should be easier to build than torch or tensorflow #50, and it would be useful to include it.
-
Depending on whether I use `jit` or not, and whether I use `jax.asarray` on numpy arrays or not, the memory usage I observe (with CPU backend) varies by a factor of four:
```
# These control the f…
-
Hi My test run failed and showed the following error messages. Can you please give me a pointer to what went wrong? Thanks.
```
Running tests under Python 3.11.10: /alphafold3_venv/bin/python
[ R…
-
Dear FLAX community,
### System information
- OS Platform and Distribution: Ubuntu 22.04.3 LTS
- Flax, jax, jaxlib versions : Flax: 0.8.1 / jax: 0.4.27 / jaxlib: 0.4.27+cuda12.cudnn89
…