All Topics
All Topics
Technology
Technology
Design
Design
Programming
Programming
Science
Science
News
News
Gaming
Gaming
Entertainment
Entertainment
Business
Business
Finance
Finance
Sports
Sports
Health
Health
Food
Food
Travel
Travel
Art
Art
Music
Music
Books
Books
Education
Education
Politics
Politics
Personal
Personal
No algorithm. No AI slop. No ads. Just RSS. Pro-human. Indie writers. Real journalism. Open web. Chronological. Hand toasted.

Microjax: A Simplified Autograd Engine with Jax-like API

By

joelburget

11mo ago· 1 min readenCode

Summary

Microjax is a tiny autograd engine with a Jax-like API inspired by Micrograd and JAX, offering a functional style for machine learning. The tutorial simplifies concepts from autograd and packages them as a notebook.

Key quotes

· 3 pulled
This is inspired by Andrej Karpathy's Micrograd, a PyTorch-like library in about 150 lines of code.
Despite PyTorch's popularity, I prefer the way JAX works because it a more functional style.
My main contribution is simplifying a bit and packaging it as a notebook.
Snippet from the RSS feed
A tiny autograd engine with a Jax-like API. Contribute to joelburget/microjax development by creating an account on GitHub.

You might also wanna read