Chủ sở hữu

jax-ml

Đã lập chỉ mục 10 kho mã nguồn · Xem trên GitHub

jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Python · 36316 star
scaling-book
Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

HTML · 1441 star
jax-triton
jax-triton contains integrations between JAX and OpenAI Triton

Python · 470 star
ml_dtypes
A stand-alone implementation of several NumPy dtype extensions used in machine learning.

C++ · 361 star
oryx
Oryx is a library for probabilistic programming and deep learning built on top of Jax.

Python · 328 star
jax-ai-stack

Python · 312 star
jax-llm-examples
Minimal yet performant LLM examples in pure JAX

Python · 281 star
bayeux
State of the art inference for your bayesian models.

Python · 248 star
bonsai
Minimal, lightweight JAX implementations of popular models.

Python · 244 star
coix
Inference Combinators in JAX

Jupyter Notebook · 54 star

Đang mở Tất cả

[BUG] BaseTpuCluster and K8sCluster coordinator discovery fails on IPv6 Kubernetes clusters due to socket.gethostbyname Đang mở

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 74/100

jax-ml/jax#40855 · Mở ngày 22/9/2026
[BUG] GkeTpuCluster queries metadata.google.internal for MEGASCALE_SLICE_ID even when TPU_SKIP_MDS_QUERY is set Đang mở

Độ khó 2/5 1-3 giờ Mức phù hợp với người mới 88/100

jax-ml/jax#40854 · Mở ngày 22/9/2026
a dynamic_slice of dynamic_slice reads the wrong element for out-of-range start indices Đang mở

bug XLA

jax-ml/jax#40849 · 2 bình luận · 1 người được giao · Mở ngày 21/9/2026
[pallas] unsound caching of `pl.num_programs()` when one kernel function is used with several grids Đang mở

bug pallas

jax-ml/jax#40847 · 1 người được giao · Mở ngày 21/9/2026
`jnp.gcd` / `jnp.lcm` never terminate when an argument is the minimum integer of its dtype (eager and jit) Đang mở

bug

jax-ml/jax#40840 · 3 bình luận · 1 người được giao · Mở ngày 21/9/2026
jax.lax.psum_scatter on CPU ignores --xla_cpu_collective_timeout_seconds and fails after 30 s Đang mở

bug

jax-ml/jax#40831 · 2 bình luận · 1 người được giao · Mở ngày 21/9/2026
jax.lax.psum_scatter on CPU intermittently returns wrong values on rank 0 Đang mở

bug

jax-ml/jax#40830 · 3 bình luận · 1 người được giao · Mở ngày 21/9/2026
lax.cond can mark an array as replicated when it is not, giving rise to silent errors with shard map Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 48/100

jax-ml/jax#40823 · 1 bình luận · Mở ngày 20/9/2026
jax.scipy.linalg.solve silently ignores assume_a='sym' and 'her' Đang mở

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 68/100

jax-ml/jax#40813 · Mở ngày 19/9/2026
JAX arrays don’t report themselves as readonly Đang mở

bug

jax-ml/jax#40789 · 2 bình luận · 1 người được giao · Mở ngày 18/9/2026
Test failures on RISC-V/musl [inverse trig functions, (b)complex32] Đang mở

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 45/100

jax-ml/ml_dtypes#404 · Mở ngày 17/9/2026
Failed build: CI - with Numpy/Scipy nightly wheels (nightly) Đang mở

build failed

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 35/100

jax-ml/jax#40753 · Mở ngày 17/9/2026
pure_callback: cross-platform sharding loses its own error message (except IndexError should be ValueError) Đang mở

Độ khó 1/5 Dưới một giờ Mức phù hợp với người mới 92/100

jax-ml/jax#40722 · 1 bình luận · Mở ngày 16/9/2026
[Pallas TPU] `jax.named_scope` splits otherwise co-issued instructions into separate bundles Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 48/100

jax-ml/jax#40720 · Mở ngày 16/9/2026
`jnp.concatenate` on minor axes significantly slower in newer versions Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 62/100

jax-ml/jax#40672 · 1 bình luận · Mở ngày 14/9/2026
betaln(a, b) is too large by exactly min(a, b) when min(a, b)/max(a, b) is subnormal Đang mở

Độ khó 2/5 1-3 giờ Mức phù hợp với người mới 76/100

jax-ml/jax#40653 · Mở ngày 14/9/2026
jax.scipy.special.spence returns nan over the top two binades of float32 and float64 Đang mở

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 72/100

jax-ml/jax#40652 · Mở ngày 14/9/2026
sph_harm_y returns incorrect values for batched angles (same root cause as #20769) Đang mở

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 65/100

jax-ml/jax#40651 · 3 bình luận · Mở ngày 14/9/2026
`jnp.arcsinh` / `jnp.arccosh` JVP is exactly 0 in float32 for $|x| > ~1.84e19$ because of squared x Đang mở

bug

jax-ml/jax#40634 · 3 bình luận · 1 người được giao · Mở ngày 11/9/2026
Per-operator (non-jitted) kernels return wrong values / NaNs on RTX A6000 (CUDA 13), jitted path and CPU are correct; breaks a jitted training loop Đang mở

Độ khó 5/5 Hơn một tuần Mức phù hợp với người mới 25/100

jax-ml/jax#40633 · 2 bình luận · Mở ngày 11/9/2026
jax.lax.scatter_max returns infinite gradient with mode='clip' and valid clipped indices Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 65/100

jax-ml/jax#40626 · Mở ngày 11/9/2026
jax.experimental.sparse.coo_matmat gives incorrect reverse derivative with transpose=True Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 64/100

jax-ml/jax#40625 · Mở ngày 11/9/2026
jax.experimental.sparse.coo_fromdense gives incorrect second derivative when nse includes padded entries Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 72/100

jax-ml/jax#40624 · Mở ngày 11/9/2026
`shape_cast` layout rule should allow rank changes for degenerate dimensions Đang mở

enhancement

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 68/100

jax-ml/jax#40610 · Mở ngày 10/9/2026
[jit] jax_high_dynamic_range_gumbel is missing from the JIT cache key Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 72/100

jax-ml/jax#40609 · 2 bình luận · Mở ngày 10/9/2026
libtpu CHECK-fails (SIGABRT) on `dot_general` with one float and one integer operand and `preferred_element_type=int32` Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 68/100

jax-ml/jax#40581 · Mở ngày 9/9/2026
GPU detected successfully but first CUDA allocation fails with CUDA_ERROR_UNKNOWN on WSL2 (RTX A4000) Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 35/100

jax-ml/jax#40558 · 6 bình luận · Mở ngày 9/9/2026
[TPU] jax 0.10.2 -> 0.11.1: +57% decode step-time regression on v7x from SparseCore gather offload at batch=1; recovered by --xla_tpu_offload_gather_to_sparsecore=false Đang mở

Độ khó 5/5 Hơn một tuần Mức phù hợp với người mới 35/100

jax-ml/jax#40553 · 3 bình luận · Mở ngày 8/9/2026
no kernel image on Jetson Orin (sm_87) with jax[cuda12] 0.11.1 aarch64 wheels Đang mở

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 42/100

jax-ml/jax#40487 · Mở ngày 6/9/2026
lax.composite lowers to invalid StableHLO under shape polymorphism and multi-platform export Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 78/100

jax-ml/jax#40486 · 3 bình luận · Mở ngày 5/9/2026
checkify: lax.select dtype error when a jax.enable_x64() scope sits inside the checkified function Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 68/100

jax-ml/jax#40482 · 1 bình luận · Mở ngày 5/9/2026
`jnp.log2` does not compute integer powers of 2 exactly Đang mở

bug

jax-ml/jax#40419 · 10 bình luận · 1 người được giao · Mở ngày 3/9/2026
XLA change significantly increase the computational time Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 48/100

jax-ml/jax#40416 · 5 bình luận · Mở ngày 3/9/2026
jax.nn.log_softmax / jax.scipy.special.log_softmax gradient returns NaN on large float16 arrays despite finite forward pass Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 55/100

jax-ml/jax#40409 · Mở ngày 2/9/2026
jax.nn.softplus second derivative returns zero for large positive float64 input Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 68/100

jax-ml/jax#40387 · 1 bình luận · Mở ngày 2/9/2026
jax.nn.softmax loses small finite float64 gradient component Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 64/100

jax-ml/jax#40386 · 1 bình luận · Mở ngày 2/9/2026
jax.nn.sigmoid gradient returns zero for large positive float64 input Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 63/100

jax-ml/jax#40385 · 1 bình luận · Mở ngày 2/9/2026
jax.nn.log_softmax loses small finite float64 gradient component Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 52/100

jax-ml/jax#40384 · 1 bình luận · Mở ngày 2/9/2026
jax.nn.log_sigmoid second derivative returns zero for finite float64 input Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 52/100

jax-ml/jax#40383 · 1 bình luận · Mở ngày 2/9/2026
jax.lax.exp2 gradient overflows to inf where finite float64 derivative is expected Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 54/100

jax-ml/jax#40382 · 1 bình luận · Mở ngày 2/9/2026
`jax.nn.softmax` default JVP violates logit-translation invariance in a finite float64 tail Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 52/100

jax-ml/jax#40344 · 1 bình luận · Mở ngày 1/9/2026
`jax.nn.sigmoid` default JVP loses a finite float64 tail derivative that JAX's highest-accuracy rule preserves Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 68/100

jax-ml/jax#40342 · 1 bình luận · Mở ngày 1/9/2026
Reverse-mode double differentiation of saturated tanh returns 0 instead of the correct second derivative Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 48/100

jax-ml/jax#40315 · Mở ngày 29/8/2026
jit lower depends on process trace history Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 50/100

jax-ml/jax#40312 · Mở ngày 29/8/2026
Aval repr is lossy. Đang mở

bug

jax-ml/jax#40288 · 2 bình luận · 1 người được giao · Mở ngày 28/8/2026
transfer_guard and jax.device_put(..., device=cpu_device) Đang mở

bug

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 58/100

jax-ml/jax#40285 · Mở ngày 28/8/2026
Programmatic profiler results Đang mở

enhancement

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 48/100

jax-ml/jax#40258 · Mở ngày 27/8/2026
All-NaN gradients from preconditioned `jax.scipy.sparse.linalg.cg` for small cotangent norms Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 55/100

jax-ml/jax#40254 · Mở ngày 27/8/2026
Shape-dependent XLA/Triton failure for jvp(grad) through CNN Đang mở

bug

Độ khó 4/5 3-5 ngày Mức phù hợp với người mới 35/100

jax-ml/jax#40237 · Mở ngày 26/8/2026
Support GMEM ranks above two in the tiled `cp.async` GMEM→SMEM copy (pre-Hopper) Đang mở

enhancement

Độ khó 3/5 1-2 ngày Mức phù hợp với người mới 78/100

jax-ml/jax#40201 · Mở ngày 25/8/2026

Đang hiển thị 100 mục mới nhất

Trang này chỉ liệt kê những gì được lập chỉ mục gần đây. Bộ lọc nâng cao có toàn bộ kho, thu hẹp theo ngôn ngữ, độ khó và thời lượng.

Mở bộ lọc nâng cao

jax-ml

Đang hiển thị 100 mục mới nhất

Nhận issue mới trong hộp thư của bạn