Defining new Jax types with hijax

fhchl 14 points 3 comments July 12, 2026
docs.jax.dev · View on Hacker News

Discussion Highlights (3 comments)

chrisjj

> the usual tool is a pytree: 404.

0xnyn

Hijacking JAX with hijax. Such a neat name.

peterdsharpe

Why use this over Equinox's Module, plus jaxtyping? That seems like the cleaner (and more established) solution, but maybe there are pros/cons I'm not seeing.

Semantic search powered by Rivestack pgvector
14,015 stories · 131,331 chunks indexed