-
The direct Swagger dep was created to have a quick and dirty hack for producing a meaningful swagger.json, b/c the controller methods just return a `Response` object, which can't be detected usefully …
-
Using a JAX device function in the host fn of `host_callback` causes hangs/long compile/oom (breakage changes with the backend).
This not being supported is probably WAI, but we could provide a bet…
-
Dear authors,
I have to do 4 dimensions reweighting for 10M events using GBDT. With the CPU only, the process takes a full day.
My question is : Is it possible to write or is there already an in…
-
Hi all,
I am wondering what is the preferred way to create a model that is too large to fit in a single device
As a reference starting point, if I use data parallelism, I will first create per-…
-
Hi,
it’s not clear how to make production level decentralized projects with Python, and the GIL is often a big limit.
Do you think it would be possible to make Jax.js, or (less ideally) use it …
-
Hi,
I would like to propose an improvement on the JIT API in order to make it closer to the vmap and pmap syntax.
At the moment functions that take a pytree as an argument either set the full py…
-
### Description
I have installed the TPU version of JAX (including jaxlib and libtpu) on all hosts of a TPU Pod inside a venv. Then, I run the following command on all hosts:
```sh
. ~/venv/bin…
-
I would like to propose addition of new `@RegisterLRAParticipant` annotation which will denote a CDI bean which will be registered as LRA participant by the LRA implementation. This annotation will be…
-
I would like to do the following, which is one step of the EM algorithm:
* set up a factor graph corresponding to a simple Bayesian network (I've got this far)
* specify evidence in the form of …
-
##### Description
It appears that the gradle open api plugin jax-rs-spec generator (java server-side api without an implementation) version 4.3.1 is not adding additional annotations on top of model …