Hacker News with Generative AI: Jax

Craftax: (Crafter and NetHack) RL Environment in Jax (github.com/MichaelTMatthews)
Craftax is an RL environment written entirely in JAX. Craftax reimplements and significantly extends the game mechanics of Crafter, taking inspiration from roguelike games such as NetHack.
Reinforcement Learning, Game Development, Python, Jax
14 points by lnyan 10 days ago | 2 comments
Shape Rotation 101: An Intro to Einsum and Jax Transformers (bearblog.dev)
Machine Learning, Programming, Linear Algebra, Transformers, Jax
130 points by dejavucoder 243 days ago | 14 comments
Maxtext: A simple, performant and scalable Jax LLM (github.com/google)
Machine Learning, Jax
115 points by zerojames 303 days ago | 7 comments