-
Notifications
You must be signed in to change notification settings - Fork 2.7k
Issues: google/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
Operator Something isn't working
jax.lax.gather
consumes a lot of memory on CPU
bug
#23457
opened Sep 5, 2024 by
daskol
Error exporting np.where when unrelated dimension variable is 0
bug
Something isn't working
#23456
opened Sep 5, 2024 by
tchatow
Multi-process GPU jobs fail on Slurm
bug
Something isn't working
#23452
opened Sep 5, 2024 by
ChenAo-Phys
TPU operations don't appear if profile is too long
bug
Something isn't working
#23446
opened Sep 5, 2024 by
jlin816
ValueError: Apply must be called with an RNG as the third argument, the required signature is: Something isn't working
apply(params, state, rng, *a, **k)
. The object was of type <class 'jax._src.interpreters.partial_eval.DynamicJaxprTracer'>: Traced<ShapedArray(uint32[2])>with<DynamicJaxprTrace(level=1/0)
bug
#23428
opened Sep 4, 2024 by
eaven123
Profile shows TPU operations before CPU instructions
bug
Something isn't working
#23426
opened Sep 4, 2024 by
jlin816
jnp.arange
does not permit dynamic shaped arrays
bug
#23423
opened Sep 3, 2024 by
josh146
CuSolver: Switch to 64 bit api to allow for eigh on matrices > than 26732x26732
enhancement
New feature or request
#23413
opened Sep 3, 2024 by
PhilipVinc
Improve CuSolver errors diagnostics
enhancement
New feature or request
#23410
opened Sep 3, 2024 by
PhilipVinc
_isolve tacitly checks for symmetric linear operator wrongly
bug
Something isn't working
#23403
opened Sep 3, 2024 by
Joshuaalbert
error: failed to legalize operation 'mhlo.erf' with jax-metal
bug
Something isn't working
#23384
opened Sep 2, 2024 by
a-gn
Segmentation Fault on 8x8 or larger matrix multiplication with GPU
bug
Something isn't working
#23355
opened Aug 30, 2024 by
treyra
jax.nn.dot_product_attention
does not respect key_value_seq_lengths
bug
#23349
opened Aug 30, 2024 by
danjenson
jnp.pow()
triggers dtype promotion when it should not
bug
#23337
opened Aug 30, 2024 by
ayaka14732
Add function that checks whether New feature or request
jax
is yet initialized or not
enhancement
#23320
opened Aug 29, 2024 by
milutter
Full zero arrays allocated for "consumed" gradients of Something isn't working
stop_gradient
parameters
bug
#23316
opened Aug 29, 2024 by
colehaus
Cannot pass constant to Something isn't working
lax.fori_loop
body inside pallas kernel
bug
#23301
opened Aug 28, 2024 by
lengstrom
Flash attention soft capping support
enhancement
New feature or request
pallas
Issues pertaining to Pallas (GPU or TPU)
#23300
opened Aug 28, 2024 by
sparsh35
jax.numpy.tan
call adds chlo.tan
call to exported StableHLO; stablehlo.tan
should be preferred
bug
#23259
opened Aug 26, 2024 by
joaospinto
sparse.BCOO error when matrix is passed as a parameter
bug
Something isn't working
duplicate
This issue or pull request already exists
#23237
opened Aug 26, 2024 by
muchanem
Wrong derivatives when using defjvp for sparse.BCOO matrices
bug
Something isn't working
#23235
opened Aug 26, 2024 by
BodeTobias
[Important] Keep non-hermetic CUDA toolkit as a build option.
bug
Something isn't working
#23231
opened Aug 26, 2024 by
yliu120
JITing loss function causes erroneous results
bug
Something isn't working
#23226
opened Aug 24, 2024 by
muchanem
Guard against unintentional transfers to CPU
enhancement
New feature or request
#23215
opened Aug 23, 2024 by
kxs-dhenshall
Rotation.concatenate does not work for two single rotations
bug
Something isn't working
#23202
opened Aug 22, 2024 by
thomas-rkk
Previous Next
ProTip!
Mix and match filters to narrow down what you’re looking for.