Microjax: 두 개의 클래스와 여섯 개의 함수로 JAX 구현

2025-07-07
Microjax: 두 개의 클래스와 여섯 개의 함수로 JAX 구현

Andrej Karpathy의 Micrograd에서 영감을 받은 Microjax는 단 두 개의 클래스와 여섯 개의 함수만으로 JAX의 기능을 재현하는 라이브러리입니다. 인기 있는 PyTorch와 달리 Microjax는 JAX의 보다 함수적인 프로그래밍 스타일을 채택합니다. 이 튜토리얼은 JAX의 전신인 autograd에 대한 Matthew J Johnson의 훌륭한 2017년 프레젠테이션을 바탕으로 간소화하여 노트북으로 패키징했습니다.

개발