Hacker News with Generative AI: Jax

Show HN: Kinematic Hand Skeleton Optimization in Jax (github.com/rerun-io)
A repo to explore training robots with Pi0 and Lerobot and human pose motion retargeting
Robotics, Machine Learning, Open Source, Jax, GitHub
20 points by pablovelagomez 21 days ago | 0 comments
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 102 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 335 days ago | 14 comments
Maxtext: A simple, performant and scalable Jax LLM (github.com/google)
Machine Learning, Jax
115 points by zerojames 394 days ago | 7 comments