-
Hi! Firstly, thanks a lot for developing this package! I'm not in astro, but the package has been just what I needed in a project I'm working on.
The issue I'm having is related to the JAX implemen…
-
### Please describe the purpose of the feature. Is it related to a problem?
Create a DreamerV3 implementation - there is no pure Jax implementation to date.
### Describe the solution you'd like
P…
-
Hello!
The following currently errors out:
```
eqx.error_if((), jnp.array(False), "Errors?")
```
with the error:
```
.venv/lib/python3.11/site-packages/equinox/_errors.py:229: in error_if
…
-
I found that for the common training pattern in Jax:
`new_state, other_output = jitted_train_step_fn(old_state, other_input)
`
Current XLA runtime may assign different backing device memory bu…
-
Your documentation [mentions](https://ydf.readthedocs.io/en/latest/#:~:text=and%20TensorFlow%20Records.-,Serving,-Benchmark%20model%20inference) :
> Run models in Python, C++, Go, JavaScript, and C…
-
For example, there is a lookup table backend, and this operation need to get the table handle to reach it. This feature is implemented in tensorflow by tf.resource.
Can jax.pure_callback and jax.ex…
-
#### Issue description
Trying to use `qml.qchem.molecular_hamiltonian` inside of a callback and it's not working.
* *Expected behavior:* I expect that generating a `Hamiltonian` in a callback sh…
-
I was wondering if it would be possible to allow [`jax.pure_callback`](https://jax.readthedocs.io/en/latest/_autosummary/jax.pure_callback.html) to infer `result_shape_dtypes` by calling the specified…
-
Is there any value in supplying a pure_callback (one with a return value), a buffer into which to place it's output? Something like the code below. I have some science applications where I need to use…
-
I have been writing pallas kernels and has recently been made aware that they are not automatically partitioned.
I guess the general case is complicated; in my own case I am going with the `custom_…
luyug updated
7 months ago