-
Notifications
You must be signed in to change notification settings - Fork 2.7k
Issues: jax-ml/jax
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Author
Label
Projects
Milestones
Assignee
Sort
Issues list
[DOC] We should document all jax config from the file config.py
enhancement
New feature or request
#23809
opened Sep 20, 2024 by
nouiz
2 tasks done
Add support for the new "dot algorithm" spec for more explicit control of dot product numerics
enhancement
New feature or request
#23797
opened Sep 20, 2024 by
dfm
Slow transpose convolutions (both cpu and cuda backends)
bug
Something isn't working
CPU
Issues related to the CPU compiler/runtime
NVIDIA GPU
Issues specific to NVIDIA GPUs
#23783
opened Sep 20, 2024 by
psmaragdis
Issue with Something isn't working
jax_getattr
inside jax.scan
when the PyTree has multiple leaves
bug
#23782
opened Sep 19, 2024 by
nelhage
Multiprocessing with Jax on Slurm fails
bug
Something isn't working
#23770
opened Sep 19, 2024 by
NonsansWD
Numerical implementation of logarithmic integral function
enhancement
New feature or request
P3 (no schedule)
We have no plan to work on this and, if it is unassigned, we would be happy to review a PR
#23732
opened Sep 18, 2024 by
Qazalbash
[jax 0.4.33] XlaRuntimeError: FAILED_PRECONDITION: Buffer Definition Event: Unsupported number of sorted inputs: 17
bug
Something isn't working
#23727
opened Sep 18, 2024 by
PhilipVinc
Typing improvement: Preserve the wrapped callable's signature in New feature or request
jax.jit
enhancement
#23719
opened Sep 18, 2024 by
lebrice
2 tasks done
cuDNN attention segmentation fault
bug
Something isn't working
NVIDIA GPU
Issues specific to NVIDIA GPUs
#23701
opened Sep 17, 2024 by
sbodenstein
OptimSteps not compatible with shard_map due to lax.cond
bug
Something isn't working
#23693
opened Sep 17, 2024 by
Rocamonde
jaxlib 0.4.32, external CUDA, gcc: error: unrecognized command-line option '--cuda-path=external/cuda_nvcc'
bug
Something isn't working
#23689
opened Sep 17, 2024 by
adamjstewart
jax.numpy.nonzero extremely slow on sharded array
bug
Something isn't working
#23675
opened Sep 16, 2024 by
muchanem
Document if differentiability of implementation is dependent on hardware
enhancement
New feature or request
#23650
opened Sep 16, 2024 by
unalmis
all_gather
in pmap
fails with new-style random keys, works with old-style
bug
#23647
opened Sep 15, 2024 by
Lookatator
When I compute gradients and perform updates using the same values in Torch and JAX, I find that after multiple iterations, the inference results differ significantly.
bug
Something isn't working
#23646
opened Sep 15, 2024 by
CZXIANGOvO
when a tracer error happens in for_loop, should point to the user's body function
better_errors
Improve the error reporting
#23637
opened Sep 13, 2024 by
mattjj
Corner-case sharding regression when replacing concrete mesh with abstract mesh
bug
Something isn't working
#23634
opened Sep 13, 2024 by
jkr26
jax.lax.linalg.lu returns LU factorisation for singular matrix, jax.scipy.linalg.lu_solve silently returns spurious result in 32bit
bug
Something isn't working
#23626
opened Sep 13, 2024 by
johannahaffner
When calculating the loss, the input data does not contain NaN, but the output contains NaN
bug
Something isn't working
#23625
opened Sep 13, 2024 by
CZXIANGOvO
pure_callback is broken with multiple vmap
bug
Something isn't working
#23624
opened Sep 13, 2024 by
Joshuaalbert
Orthogonal Initializer raises gpusolverDnCreate(&handle) failed: cuSolver internal error
bug
Something isn't working
#23616
opened Sep 13, 2024 by
brianorbrain
Make jax.distributed timeouts configurable via jax.config
enhancement
New feature or request
#23599
opened Sep 12, 2024 by
skye
Allow tuple inputs to New feature or request
scatter_dimension
in jax.lax.psum_scatter
enhancement
#23594
opened Sep 12, 2024 by
vvvm23
Sparse matrix multiplication and output nse values
enhancement
New feature or request
#23576
opened Sep 11, 2024 by
joaospinto
Error compile: --bazel_options=--repo_env=LOCAL_CUDA_PATH="${CUDA_HOME}" issues with clang & gcc
bug
Something isn't working
#23575
opened Sep 11, 2024 by
benkirk
Previous Next
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.