Skip to content

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.

Sign up for GitHub

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

1,489 Open 4,428 Closed
1,489 Open 4,428 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Issues list

TSAN race in //tests:export_back_compat_test_cpu under free-threading bug Something isn't working free threading Issues found in free threading builds
#26272 opened Feb 3, 2025 by hawkinsp
@vfdev-5
Build fails due to missing visibility declaration for hipfft target AMD GPU Issues pertaining to AMD GPUs (ROCM) bug Something isn't working build
#26269 opened Feb 3, 2025 by TimoStolz
@JehandadKhan
2
Grouped convolutions are much slower than standard convolutions when increasing kernel dilation bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs performance make things lean and fast
#26266 opened Feb 2, 2025 by karunrao97
@mjsML
1
Support Bounded Shape Input enhancement New feature or request
#26265 opened Feb 2, 2025 by yaochengji
@mattjj
Gradient of jnp.linalg.norm does not respect absolute homogeneity. bug Something isn't working
#26248 opened Jan 31, 2025 by tillahoffmann
@mattjj
1
jax.ops.segment_sum is 100x slower when segment_ids are sorted, but only for float16 bug Something isn't working NVIDIA GPU Issues specific to NVIDIA GPUs
#26227 opened Jan 31, 2025 by rafaelha
@pifon2a
1
Document jax.lax APIs documentation
#26219 opened Jan 30, 2025 by jakevdp
@jakevdp
Control the lifetime of host values of the jax.Array. bug Something isn't working
#26216 opened Jan 30, 2025 by yliu120
@dfm
10
Document more practical use-cases for vmap and cases that cannot be covered by additional batch axes enhancement New feature or request
#26212 opened Jan 30, 2025 by adonath
@jakevdp
1
Issues building from source on an M3 chip bug Something isn't working
#26207 opened Jan 30, 2025 by botev-openai
6
Jaxpr string representation of dataclasses enhancement New feature or request
#26202 opened Jan 30, 2025 by botev
@jakevdp
1
Add documentation about behavior of linear solvers in case the system matrix is singular enhancement New feature or request
#26180 opened Jan 29, 2025 by NKlug
2
Operation very slow to compile since jax 0.4.36 bug Something isn't working
#26162 opened Jan 28, 2025 by btaba
7
Performance degradation of map_coordinates with the new CPU runtime bug Something isn't working
#26152 opened Jan 28, 2025 by rifu4
2
shard_map errors when composing with jax.jacrev or jax.jacfwd with jax 0.5.0 bug Something isn't working
#26148 opened Jan 28, 2025 by mganahl
CPU identity subtraction < 0 bug Something isn't working
#26147 opened Jan 28, 2025 by andrewlkd
8
CPU slowdown with new runtime (v0.4.32 and newer) bug Something isn't working
#26145 opened Jan 28, 2025 by michaeldeistler
4
jax.experimental.mosaic.gpu.profiler.measure is missing a warmup? bug Something isn't working
#26144 opened Jan 28, 2025 by mariogeiger
Jax 0.5.0 error: no member named 'Auto' in 'llvm::ms_demangle::PrimitiveKind' bug Something isn't working
#26143 opened Jan 28, 2025 by gongomgra
@superbobry
1
Pallas test failures on CPU on Intel Emerald Rapids bug Something isn't working CPU Issues related to the CPU compiler/runtime
#26130 opened Jan 27, 2025 by hawkinsp
@justinjfu
Make linear transpose use backward pass of custom vjp enhancement New feature or request
#26121 opened Jan 27, 2025 by Magwos
5
tests/pjit_test.py::PJitErrorTest::testAxisResourcesMismatch fails with "SystemError: nanobind::detail::nb_func_error_except(): exception could not be translated!" bug Something isn't working
#26106 opened Jan 26, 2025 by booxter
Exceptions from jax.pure_callback have no __cause__ bug Something isn't working
#26102 opened Jan 25, 2025 by crusaderky
16
printing new style prng keys on the cpu results in XlaRuntimeError bug Something isn't working
#26101 opened Jan 25, 2025 by SimonKoop
@jakevdp
10
Suppress XLA slow operation alarm enhancement New feature or request
#26098 opened Jan 25, 2025 by cool-RR
ProTip! Add no:assignee to see everything that’s not assigned.