서평
-
[리뷰] JAX/Flax로 딥러닝 레벨업서평 2024. 10. 2. 20:48
딥러닝 업계는 매우 빠르게 변한다. 이는 심지어 모델 뿐만이 아니라 사용되는 프레임워크에도 통하는 이야기이다. 중학교 때 처음 scikit-learn을 배워 글자 인식 인공지능을 만들어보고, 케라스가 나왔을 때에는 이렇게 편한 프레임워크를 두고 설마 다른 프레임워크가 나오겠냐는 생각을 했었다. 그러나, 오늘날은 PyTorch의 세상이다. 그리고, Jax가 다시 한번 이 세상을 뒤집으려 하고 있다. 지난 학기에 친구의 손에 이끌려 인공지능 개론이라는 수업을 수강했다. 딥러닝 모델 중 하나를 선택해 이를 개선해보는 프로젝트형 과목이었는데, 우리는 DreamerV3이라는 WM 프로젝트를 Jax를 이용하여 구현하는 프로젝트를 진행했다. 이러한 경험 때문에 서평단을 신청해보게 되었다. Jax를 한마디로 요약하자면..