-
Hi there,
Thank you very much for putting this package together, this is impressive! I was wondering if you would be interested in an implementation of nested sampling in pure jax. I know that the […
-
Hi guys,
I am using a JAX-based framework to do finite element analysis.
Currently, I use solvers from two sources; either `jax.scipy.bicgstab` or PETSc.
I like the fact that `lineax` has many so…
-
Right now, some quantities like `.radius` act like attributes that don't need to be called with a `()`, and others like `.stellar_brightness(wavelength=...)` act like methods that must be called with …
-
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
…
-
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_…
-
# 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 months ago
-
I'm trying to add a little wrapper to a DeviceArray to handle [astropy.units](https://docs.astropy.org/en/stable/units/), which are widely used in astronomy code. I can't figure out how to make this c…
-
The benefit of using JAX is the ability to JIT compile. With the setup right now, it is not clear what's the best way to make JAX recognize QuTiP objects as valid inputs since JIT only works for pure …
-
Hi,
how may I obtain the predictions of the World Model during Inference?
I have tried this command in a simple inference loop, but it throws an error: `agent.agent.wm.imagine(agent.policy, obs, 1…
-
## Project description
At the moment there are various flavors for compilers and stdenvs in `llvmPackages`. I believe most of the existing configurations are not very useful and the most important …