Logo Questions Linux Laravel Mysql Ubuntu Git Menu
HTML CSS JAVASCRIPT SQL PYTHON PHP BOOTSTRAP JAVA JQUERY R React Kotlin
 

New posts in jax

What are the tradeoffs between jax.lax.map and jax.vmap?

python jax

JAX, recompilation when using closure for a function

python jax

JAX vs NumPy array indexing - out of bounds behaviour

python-3.x numpy multidimensional-array jax

Jax, jit and dynamic shapes: a regression from Tensorflow?

python tensorflow jax

in_axes keyword in JAX's vmap

jax auto-vectorization

How to select between using a `jax.lax.scan` vs a `for` loop when using JAX?

python jax

How to Make Batching Rule for Multiple Outputs

python jax

JAX scan over a leading dimension (normal way) or with an index

python jax

Jax - Debugging NaN-values

python nan recurrent-neural-network jax

is it possible to jit a function which uses jax.numpy.unique?

jit jax

Apply function only on slice of array under jit

python numpy jax

JAX: Why not using @jit yields -inf value but using it doesn't?

python jax

Handle varying shapes in jax numpy arrays (jit compatible)

python arrays shapes jit jax

Jax vectorization: vmap and/or numpy.vectorize?

python jax

Stable diffusion: AttributeError: module 'jax.random' has no attribute 'KeyArray'

python google-colaboratory jax stable-diffusion

How do I save an optimizer state of JAX trained model?

python machine-learning jax

Jaxlib pip installation failure

python windows pip jax

Vectorize jax.lax.cond with vmap

python numpy numpy-ndarray jax autodiff

Test jax.pmap before deploying on multi-device hardware

multiprocessing tpu jax pmap

Nontransitive subclassing with numpy and jax

python numpy jax
Older Entries »