-
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
Persistent compilation cache hit but tracing cache miss?
bug
Something isn't working
#22281
opened Jul 4, 2024 by
colehaus
Human readable New feature or request
as_text
for jax.stages.Compiled
/ HloModule
enhancement
#22270
opened Jul 4, 2024 by
balancap
Implement QR decomposition derivative in complete mode
enhancement
New feature or request
#22264
opened Jul 3, 2024 by
JGameCreation
Fine-grained remat policy makes async/pipelined collectives execute in the main stream
bug
Something isn't working
#22252
opened Jul 3, 2024 by
qGentry
Incorrect cost_analysis() Flop Count on GPU for Dense layer/Dot product
bug
Something isn't working
#22249
opened Jul 3, 2024 by
m-schier
Allow overloading binary operations between jax.Array and custom array-like types (perhaps via New feature or request
__array_priority__
)
enhancement
#22240
opened Jul 2, 2024 by
danieldjohnson
Feature Request: a way to opt out of "hermetic python"
enhancement
New feature or request
#22216
opened Jul 1, 2024 by
SomeoneSerge
2 tasks done
Check failed in collective_pipeliner when using gradient accumulation with non-unrolled loop
bug
Something isn't working
type:Bug
#22210
opened Jul 1, 2024 by
qGentry
pallas hello world not implemented for int8
bug
Something isn't working
pallas
Issues pertaining to Pallas (GPU or TPU)
#22204
opened Jun 30, 2024 by
vlad17
Automatically differentiate Pallas kernel failed
bug
Something isn't working
pallas
Issues pertaining to Pallas (GPU or TPU)
#22197
opened Jun 29, 2024 by
Lime-Cakes
Efficient parallelization of tasks with highly variable difficulty and run times
enhancement
New feature or request
#22176
opened Jun 28, 2024 by
javier-garcia-tilburg
2 tasks done
Error Building Jaxlib v0.4.30 on Jetson Orin
bug
Something isn't working
#22155
opened Jun 27, 2024 by
jwknaup
How to perform a Bucketize and dynamic_partition OP?
question
Questions for the JAX team
#22151
opened Jun 27, 2024 by
MoFHeka
2 tasks done
Global Singleton (XlaDebugInfoManager) leaks out of the control of C API and gets two copies in two shared libraries
bug
Something isn't working
#22148
opened Jun 27, 2024 by
yliu120
Wrong result for unsigned dtype input into Something isn't working
jax.numpy.partition
and jax.numpy.argpartition
bug
#22137
opened Jun 27, 2024 by
JuliaPoo
cuSolver internal error with Something isn't working
jax.scipy.stats.multivariate_normal.pdf()
bug
#22127
opened Jun 26, 2024 by
jwknaup
int4 reshape: Reshape should have supported layout before reaching the emitter
bug
Something isn't working
#22121
opened Jun 26, 2024 by
py4
Is there difference in computing gradient between jax and torch
bug
Something isn't working
#22101
opened Jun 26, 2024 by
ergo-zyh
Figure out + document how to use manual capture profiling with Kubernetes
documentation
enhancement
New feature or request
#22096
opened Jun 25, 2024 by
skye
Array dispatching with __array_ufunc__ in JAX
enhancement
New feature or request
#22090
opened Jun 25, 2024 by
gautierronan
jaxlib or libtpu not detected on TPU Pod
bug
Something isn't working
#22070
opened Jun 24, 2024 by
ayaka14732
scan of device_put carry raises TypeError during the backward pass
bug
Something isn't working
#22045
opened Jun 22, 2024 by
fehiepsi
jaxlib wheel hardcoded to manylinux2014 platform
bug
Something isn't working
#22034
opened Jun 21, 2024 by
mrodden
lax.cond
sometimes inserts a nonlinear lax.stop_gradient
into its JVP rule.
bug
#22011
opened Jun 20, 2024 by
patrick-kidger
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.