Defining new Jax types with hijax

ffhchl about 3 hours ago 2 commentsRead Article on docs.jax.dev

FR version is available. Content is displayed in original English for accuracy.

Discussion (2 Comments)Read Original on HackerNews

0xnyn•about 2 hours ago

Hijacking JAX with hijax. Such a neat name.

chrisjj•about 2 hours ago

> the usual tool is a pytree:

404.