Showh HN: Microjax – JAX in two classes and six functions Posted by joelburget on Monday July 07, 2025 @09:41AM from the github.com dept. https://github.com/joelburget/microjax ( 1 comment | 25 points | on HN )
https://docs.jax.dev/en/latest/autodidax.html