-
Is there any plan to support the log corresponding to PrintCompilation in the future?
-
- [x] graph-schedule is O(N^2)
- [ ] group->schedule has id->users, that should be ridiculously slow. so cache it
-
### Description
I want to save the executable file generated by jax.jit in the main process and execute this executable file in another process. However, I have found that the performance of the exec…
-
See https://github.com/paradigmxyz/alphanet/issues/86#issuecomment-2377980590
-
When I jit a function that contains nap.compute_wavelet_transform, I got error:
The sig I'm feeding in is a Tsd, so I wonder if it's some issue with jax.
-
### System Info
macos/linux cpu/nvidia gpu
### Who can help?
_No response_
### Information
- [ ] The official example scripts
- [ ] My own modified scripts
### Tasks
- [ ] An off…
-
Versions:
- CUDA 12.2
- Debian 11
- triton==3.1.0
- torch==2.1.2
I have a kernel where in one particular line I define a tensor of -inf. This is all in the body of a function decorated with tri…
-
Hello，
I tried to enbale the JIT of mozjs115 by setting `CONFIGURE_FLAGS += --enable-jit` in `debian/rules`
However, after making this change, there is 390 failures reported.
Could you provide so…
-
diffrax 0.6.0
jax 0.4.30
Using jax.experimental.enable_x64 and jit will return a exception.
```python
import jax
import jax.numpy as jnp
import diffrax
key = jax.random.PRNGKey(0)
x1 = …
-
If going through the motions to reinvent the wheel and already having access to a parser, why not make it uniquely advantageous and practical?
One of the biggest problems with bash and traditional …