MicroJax
MicroJax is a minimal implementation of JAX in pure Python. It is a fun project to learn about JAX and how it works under the hood.
MicroJax is a minimal implementation of JAX in pure Python. It is a fun project to learn about JAX and how it works under the hood.