jax-ml/jax issues and pull requests

#14019 - jax.scipy.linalg.eigh_tridiagonal() doesn't implement calculation of eigenvectors

Issue - State: open - Opened by crowsonkb almost 2 years ago - 9 comments
Labels: enhancement

#13680 - Support scipy.special.lambertw

Issue - State: open - Opened by DyeKuu almost 2 years ago - 2 comments
Labels: enhancement, P3 (no schedule)

#12494 - Inverse Accumulation Mode

Issue - State: open - Opened by NeilGirdhar about 2 years ago - 10 comments
Labels: enhancement

#12191 - Deleting class instance doesn't free memory used by jitted method

Issue - State: open - Opened by f0uriest about 2 years ago - 1 comment
Labels: bug

#10192 - 'libtpu.so already in use' but actually not used

Issue - State: open - Opened by ayaka14732 over 2 years ago - 12 comments
Labels: bug

#10070 - Add environment variable for setting logging level

Issue - State: open - Opened by skye over 2 years ago - 4 comments
Labels: enhancement, P3 (no schedule)

#9259 - Jax GMRES on GPU largely slower than its scipy counterpart

Issue - State: open - Opened by Azercoco almost 3 years ago - 15 comments
Labels: enhancement, performance, XLA, NVIDIA GPU

#8807 - Implement quasi-random generators for JAX

Issue - State: closed - Opened by AdrienCorenflos almost 3 years ago - 11 comments
Labels: enhancement

#6853 - Need a way to specify constants

Issue - State: closed - Opened by remifan over 3 years ago - 11 comments
Labels: question

#6221 - Provide better error message when using device_put_sharded inside jitted code

Issue - State: open - Opened by GeorgOstrovski over 3 years ago - 2 comments
Labels: bug

#5178 - cond of pmap results in internal assertion error

Issue - State: closed - Opened by gnecula almost 4 years ago - 7 comments
Labels: bug

#3694 - JAX Profiling in Colab

Issue - State: open - Opened by 8bitmp3 over 4 years ago - 9 comments
Labels: documentation, contributions welcome, P2 (eventual)

#3112 - sparse eigenvalue solvers

Issue - State: open - Opened by mganahl over 4 years ago - 50 comments
Labels: enhancement, contributions welcome

#2748 - support jax.numpy.linalg.eig eigenvector derivatives

Issue - State: open - Opened by froystig over 4 years ago - 53 comments
Labels: enhancement

#1937 - Is there equivalent of torch.no_grad?

Issue - State: closed - Opened by HHalva almost 5 years ago - 10 comments
Labels: question

#743 - Limit jax multithreading

Issue - State: closed - Opened by samuela over 5 years ago - 12 comments
Labels: enhancement

Ecosyste.ms: Issues

GitHub / jax-ml/jax issues and pull requests