-
Dear iminuit developers,
Thank you very much for this great package!
I am the author of [evermore](https://github.com/pfackeldey/evermore) - a pure JAX based package to build binned likelihoods…
-
I am trying to keep a count of how many times the `__call__` method of a pytree is compiled and what's the size of the the cache. For a jitted function in pure jax, we can simply check the cache size …
-
**Is your feature request related to a problem? Please describe.**
The LRGS main interface is already an HTML page. Though well embedded into Java. A pure HTTP(s) interface would simplify a lot of op…
-
Hello, my lab would be interested in porting Stratego to use in pure end-to-end jax rl. Is there any interest in collaborating on this?
renos updated
6 months ago
-
Hi,
I tried implementing a constraint on the max current in a coil. The objective looks like this:
```
def current_penalty_pure(I, threshold):
return jnp.maximum(abs(I) - threshold, 0)**2
…
-
Currently, it is impossible to use `jax.lax.pswapaxes` on tensors that don't either come directly from the input or are the result of computation from the inputs.
Examples:
This works fine
```p…
-
# Dynamo backend for torchxla2
## Goal
Have a dynamo backend backend by torch_xla2.
The users should be able to do the following:
```python
m = model ...
m_compiled = torch.compile(m, ba…
qihqi updated
3 weeks ago
-
Dear jax team,
this is just a friendly bump on the implementation of eigendecomposition and batched SVD on GPU. Are you planning on implementing these?
Should I want to implement it myself, wou…
-
See gh-18286 for the proposed design changes, and gh-18668 for the main PR that laid the foundations (it supported all of `scipy.cluster`, added a CI job that used both `pytorch-cpu` and `numpy.array_…
-
### Description
`jax.closure_convert` applied to a vjp closure triggers an empty `AssertionError` as if there is a mismatch in the input trees. Code for reproduction:
```
import jax
import jax.n…