Issues

• optax: schedule-based transforms still can't run inside jit (count.item() in scaleBySchedule / addDecayedWeights)

  ekzhang/jax-js#242 · Filter by author YidaWeng opened on Oct 26, 2026
  3

• jit compile time doubles per step when an op reuses its operands in swapped order

  ekzhang/jax-js#233 · Filter by author andrinr opened on Oct 17, 2026

• Inference engine

  ekzhang/jax-js#213 · Filter by author ekzhang opened on Sep 4, 2026

• Bug: isnan() faulty with Deno WebGPU

  ekzhang/jax-js#85 · Filter by author olli4 opened on Feb 22, 2026
  9

• Leak checker

  ekzhang/jax-js#81 · Filter by author olli4 opened on Feb 10, 2026

• Faster streaming attention

  ekzhang/jax-js#66 · Filter by author ekzhang opened on Jan 28, 2026
  1

• jax.lax.scan

  ekzhang/jax-js#39 · Filter by author jimfleming opened on Dec 31, 2025
  4

• 🎁 Request a new function

  ekzhang/jax-js#26 · Filter by author ekzhang opened on Dec 29, 2025
  3