jax-CodePudding

Tags > jax

09-11MobileHow to map the kronecker product along array dimensions?
08-30OSJAX: Getting rid of zero-gradient
08-26Back-endRecursion in JAX
07-09NetIs it safe to read the value of numpy.empty or jax.numpy.empty?
06-09Software designModify an array from indexes contained in another array
05-29Software designGet dictionary keys for given batched values - Python
05-26OSCheck if 2D sub-array is ordered - Pyhthon JAX
05-10Software designautomate code conversion from one ML framework to another
04-23otherRepeating rows from array
04-23databaseJAX pytree for xarray
04-20OSinserting rows and columns of zeros to a sparse array in python
04-05Software designHow to check if a value is in an array while using jax
03-08Back-endMask a numpy array after a given value
02-14EnterpriseMerge dataclasses in python
02-11databaseswapaxes and how it is implemented?
12-18front endFastest way to multiply and sum 4D array with 2D array in python?
12-15MobileIs there a module to convert a tensorflow NN to Jax?
12-06NetAutomatic Differentiation with respect to rank-based computations
11-30databasePython - time difference (JAX library)
11-27NetJAX: avoid just-in-time recompilation for a function evaluated with a varying number of elements alo
11-27NetJAX hook / information / warning when a JIT function is re-compiled
11-05otherVectorise nested vmap
11-05Software designJax/Flax (very) slow RNN-forward-pass compared to pyTorch?
11-01databaseTurn a tf.data.Dataset to a jax.numpy iterator
10-26OSjax woes (on an NVDIA DGX box, no less)
10-16MobileHow to use grad convolution in google-jax?
10-05Software engineeringHow to use jax vmap for nested loops?