Trax — 깔끔한 코드와 속도를 위한 딥러닝 라이브러리
Trax
딥러닝 모델을 "읽기 좋은 코드로" 빠르게 만들고 싶다면 Trax를 볼 만해요. Trax는 깔끔한 코드와 속도에 집중한 end-to-end 딥러닝 라이브러리로, Google Brain 팀이 적극적으로 쓰고 유지 관리해요. 기본 모델(ResNet·LSTM·Transformer)과 RL 알고리즘(REINFORCE·A2C·PPO), 연구용 새 모델(Reformer), AWR 같은 새 RL 알고리즘을 담고 있어요. trax.fastmath의 백엔드로 JAX와 TensorFlow NumPy를 쓰며 CPU·GPU·TPU에서 그대로 동작해요.
이 카테고리의 문서
- 소개: Trax 개요와 사전학습 Transformer로 번역기 만들기
- 빠른 시작: 텐서·fastmath·레이어·데이터 파이프라인
- 문서·API: 튜토리얼과 API 인덱스