Skip to content

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.

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,364 Open 3,969 Closed
1,364 Open 3,969 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

DCT normalization affects distribution of spectral coefficients
#22476 opened Jul 16, 2024 by unalmis
Wrong normalization in FFT methods bug Something isn't working type:Bug
#22466 opened Jul 15, 2024 by unalmis
@mattjj
4
Pipelined model(1F1B pipeline parallelism) implementation in Jax. enhancement New feature or request
#22438 opened Jul 15, 2024 by MoFHeka
2 tasks done
Deadlock When Running distributed.initialize() on one TPU Host by Accident better_errors Improve the error reporting
#22436 opened Jul 14, 2024 by s-smits
1
[pallas] Documentation for tpu compiler options enhancement New feature or request pallas Issues pertaining to Pallas (GPU or TPU)
#22435 opened Jul 14, 2024 by Lime-Cakes
@justinjfu
How can I implement a custom operations with stateful resource? enhancement New feature or request
#22412 opened Jul 12, 2024 by MoFHeka
2 tasks done
6
Accuracy of jax.experimental.sparse.spsolve question Questions for the JAX team
#22399 opened Jul 11, 2024 by marcdelabarrera
3
Incorrect Jacobian of jax.scipy.special.logsumexp when b contains a 0 bug Something isn't working
#22398 opened Jul 11, 2024 by Adrien-Kahn
@superbobry
1
Local shards accessed using devices_indices_map are scaled by # of pods bug Something isn't working
#22387 opened Jul 11, 2024 by arvoelke
2
Accidentally exponential-time analysis pass in compiler? bug Something isn't working
#22385 opened Jul 11, 2024 by kach
9
Getting different results from CPU vs. CUDA backend bug Something isn't working
#22382 opened Jul 10, 2024 by dionhaefner
3
TPUVM program hangs when using jax_compilation_cache_dir with wrong gcs directory bug Something isn't working
#22364 opened Jul 10, 2024 by entrpn
@ayaka14732
8
Support dlpack pinned host memory device type (kDLCPUPinned / kDLCUDAHost) bug Something isn't working
#22346 opened Jul 9, 2024 by JesseFarebro
@superbobry
Profiler fails on kaggle tpu v3-8 notebook bug Something isn't working
#22294 opened Jul 5, 2024 by Lime-Cakes
4
Human readable as_text for jax.stages.Compiled / HloModule enhancement New feature or request
#22270 opened Jul 4, 2024 by balancap
2
Implement QR decomposition derivative in complete mode enhancement New feature or request
#22264 opened Jul 3, 2024 by JGameCreation
@mattjj
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
@mattjj
2
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
@mattjj
1
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
2
Check failed in collective_pipeliner when using gradient accumulation with non-unrolled loop bug Something isn't working
#22210 opened Jul 1, 2024 by qGentry
@mattjj
2
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
@superbobry
4
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
4
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
2
ProTip! What’s not been updated in a month: updated:<2024-06-16.