https://joey00072.com/blog A blog about AI, Python, and other things I find interesting. en-us 00shxf@gmail.com (Joey00072) 00shxf@gmail.com (Joey00072) Mon, 26 Aug 2024 00:00:00 GMT https://joey00072.com/blog/microjax https://joey00072.com/blog/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. Mon, 26 Aug 2024 00:00:00 GMT 00shxf@gmail.com (Joey00072) jaxmicrojaxtpupytorchflaxjax from scratch