-
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
pallas hello world not implemented for int8
bug
Something isn't working
#22204
opened Jun 30, 2024 by
vlad17
Automatically differentiate Pallas kernel failed
bug
Something isn't working
#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
Mark jax typing Protocols as New feature or request
@runtime_checkable
enhancement
#22144
opened Jun 27, 2024 by
bionicles
2 tasks done
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
jax.xla_computation as_hlo_text()
enhancement
New feature or request
#21990
opened Jun 20, 2024 by
aoshujiaocheng
T5x+jax pretraining longt5xl-3B 4096 input oom
bug
Something isn't working
#21985
opened Jun 20, 2024 by
robotzheng
Import error
bug
Something isn't working
needs info
More information is required to diagnose & prioritize the issue.
#21940
opened Jun 18, 2024 by
jntdst
Docs for persistent compilation cache are superseded
bug
Something isn't working
documentation
#21908
opened Jun 17, 2024 by
epignatelli
Is there a way to control the pjit to dynamically partition the input IDs into different ranks?
needs info
More information is required to diagnose & prioritize the issue.
question
Questions for the JAX team
#21902
opened Jun 16, 2024 by
MoFHeka
2 tasks done
Edge behavior in Something isn't working
jax.scipy.special.betainc
bug
#21900
opened Jun 15, 2024 by
mdhaber
forward and reverse mode gradients for eigvalsh + log or eigvalsh + sqrt returns NaNs
bug
Something isn't working
#21896
opened Jun 15, 2024 by
h-vijayakumaran
Previous Next
ProTip!
Type g p on any issue or pull request to go back to the pull request listing page.