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

New posts in jax

Purpose of stop gradient in `jax.nn.softmax`?

python machine-learning deep-learning autograd jax

Find gradient of a function: Sympy vs. Jax

python sympy autodiff jax

Why is array manipulation in JAX much slower?

python arrays numpy performance jax

How to reduce JAX compile time when using for loop?

python performance loops jit jax

How to vectorize a function over a list of unequal length arrays in JAX

python vectorization jax

What is the recommended way to do embeddings in jax?

jax

vmap in Jax to loop over arguments

python arrays jax

Updating entire row or column of a 2D array in JAX

python matrix multidimensional-array jax

How to use JAX pmap with CPU cores

jax

How to generate random numbers between 0 and 1 in jax?

jax

How to slice jax arrays using jax tracer?

jax

Efficient way to compute Jacobian x Jacobian.T

python pytorch backpropagation autograd jax

How to use Jax vmap over zipped arguments?

python jax

Execute function specifically on CPU in Jax

python memory cpu tpu jax

Is it safe to read the value of numpy.empty or jax.numpy.empty?

c numpy memory language-lawyer jax

Unable to Install Specific JAX jaxlib GPU version

installation gpu failed-installation jax

How to improve Julia's performance using just in time compilation (JIT)

python optimization julia jax

How to make a function a valid jax type?

jax

Jax vmap limit memory

python jax

Installing jaxlib for cuda 11.8

ubuntu pip jax
Older Entries »