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

New posts in 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

Google JAX 1D convolutional neural network

python conv-neural-network jax

JAX: JIT compatible sparse matrix slicing

python numpy jax

JAX pmap with multi-core CPU

multicore jax pmap

Performance drop when slicing jax.numpy arrays

python performance scipy jit jax

JAX vmap vs pmap vs Python multiprocessing

parallel-processing multiprocessing python-multiprocessing jax spmd

importing jax fails on mac with m1 chip

python python-3.8 apple-m1 jax

Understanding JAX argnums parameter in its gradient function

python automatic-differentiation jax

Why does GPU memory increase when recreating and reassigning a JAX numpy array to the same variable name?

python memory gpu nvidia jax

TypeError: unhashable type: 'ArrayImpl' when trying to use Equinox module with jax.lax.scan

python jax equinox computation-graph
Older Entries »