patrick-kidger/quax

Issues

• How to implement jax.lax.while with quax

  #15 opened 2 months ago
  2

• quax boundary recursive wrapping of a pytree

  #13 opened 3 months ago
  2

• Compatibility with `jaxtyping`

  #12 opened 4 months ago
  1

• jit and vamp best practices

  #11 opened 5 months ago
  1

• Interest in add array-api module

  #10 opened 5 months ago
  13

• Unitful example

  #8 opened 5 months ago
  1

• `quaxify` on a `jax.grad`

  #5 opened 5 months ago
  10

• Support for `zeros_like` and related

  #2 opened 5 months ago
  9