{"full_name":"jax-ml/jax","owner":"jax-ml","name":"jax","url":"https://github.com/jax-ml/jax","description":"Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more","homepage":"https://docs.jax.dev","topics":[],"language":"Python","license":"Apache-2.0","avatar":"https://avatars.githubusercontent.com/u/58486408?v=4","created_at":"2018-10-25T21:25:02Z","pushed_at":"2026-11-10T09:28:01Z","metrics":{"stars":36596,"forks":3997,"watchers":328,"issues":7449,"prs":31444,"releases":143,"issues_open":null,"issues_closed":null,"prs_open":null,"prs_closed":null,"prs_merged":null,"discussions":2957},"growth":{"day":{"stars":10,"forks":4,"base_day":"2026-11-09"}},"history":{"dates":["2026-11-09","2026-11-10"],"stars":[36586,36596],"forks":[3993,3997],"watchers":[328,328],"issues_open":[null,null],"prs_open":[null,null]},"boards":{},"same_owner":[],"issues":[{"number":4493,"title":"Building on linux with ppc64le CPU","comments":76,"created_at":"2020-10-08T05:48:14Z"},{"number":23689,"title":"Build issues with local CUDA installation","comments":58,"created_at":"2024-09-17T09:52:46Z"},{"number":3112,"title":"sparse eigenvalue solvers","comments":50,"created_at":"2020-05-15T22:26:23Z"},{"number":1032,"title":"feature request: sparse jacobian and sparse hessians","comments":43,"created_at":"2019-07-18T22:27:27Z"},{"number":669,"title":"Support autodiff of Eigendecomposition with repeated eigenvalues","comments":39,"created_at":"2019-05-04T06:48:40Z"},{"number":26000,"title":"Missing `.device` attribute inside `@jax.jit`","comments":38,"created_at":"2025-01-21T08:53:34Z"},{"number":1531,"title":"Differentiable matrix-free linear algebra, optimization and equation solving","comments":38,"created_at":"2019-10-20T19:46:04Z"},{"number":24411,"title":"JAX code is extremely slow on GPUs","comments":37,"created_at":"2024-10-20T16:00:25Z"}],"discussions":[{"number":42116,"title":"[Pallas] Fused forward+backward kernel for Gated DeltaNet-2 on TPU: 411× over jax.lax.associative_scan, and open questions on head-sharding","comments":0,"category":"Show and tell","created_at":"2026-10-29T18:00:04Z"},{"number":41871,"title":"Experimenting with Threefry and Pallas on TPU","comments":0,"category":"Show and tell","created_at":"2026-10-21T21:08:33Z"},{"number":41673,"title":"Failed to allocate device memory in Neural GCM simulation","comments":1,"category":"Q&A","created_at":"2026-10-16T08:31:38Z"},{"number":41651,"title":"jaxviz - Package to interactively visualize Jax models in notebooks","comments":0,"category":"Show and tell","created_at":"2026-10-15T21:31:32Z"},{"number":41384,"title":"Extending roofline","comments":2,"category":"Ideas","created_at":"2026-10-08T16:34:49Z"},{"number":41167,"title":"Recipe for automatically plumbing PRNG keys","comments":1,"category":"Ideas","created_at":"2026-10-01T02:10:11Z"},{"number":41095,"title":"What was Jax's idea behind implementing the C standard Annex G recoveries","comments":1,"category":"General","created_at":"2026-09-29T12:46:18Z"},{"number":41045,"title":"Development strategies for dealing with backend changes in XLA","comments":2,"category":"Q&A","created_at":"2026-09-28T14:02:05Z"}],"latest_release":{"name":"JAX v0.12.0.1","tag":"jax-v0.12.0.1","published_at":"2026-10-22T05:37:05Z","body":"* Bug fixes\r\n  * Built C++-generated intrinsic LLVM IR on Windows with `clang-cl` instead of\r\n    emitting empty bitcode, enabling vectorized CPU intrinsics on Windows\r\n    (https://github.com/openxla/xla/commit/3af77b5776fbc41a65d54876cc837bdc57540005)."},"readme":true,"readme_zh":false,"news":[],"stale":false,"snapshot_at":"2026-11-10T09:35:25Z","history_start":"2026-09-06"}