jax-experimental

jax-experimental copied to clipboard

Remove Run on Gradient links

1

Remove Run on Gradient links as contract with Paperspace has expired

JamesRandom

Add a breadcrumb for custom ops to performance doc.

awf

### Description The `jax.debug.callback` IPU unit tests is blocking when ignoring the returned value of the jitted function. Probably a deadlock in the IPU PjRt client at shutdown (buffer deleted?)....

balancap

BUG: jnp.cumsum(np.arange(2**14)) gives segment fault.

2

### Description Reproducer ``` import numpy as np import jax import jax.numpy as jnp # works #range = np.arange(2**13) #print(np.cumsum(range)) #print(jax.jit(jnp.cumsum, backend="ipu")(range)) # gives segment fault range = np.arange(2**14) print(np.cumsum(range))...

AlexanderMath

It would be great to be able to run e.g. the MNIST example notebook on Paperspace with a "Run on gradient" button.

awf

`import jax` is crashing with a cryptic message when `jaxlib` version does not coincide with the local Poplar SD (see #12 ) We should implement a mechanism for early catching...

balancap