파이트리
JAX에서 사용하는 트리 구조의 데이터 컨테이너로, 딕셔너리나 리스트 등이 중첩된 구조를 의미한다. 복잡한 모델 파라미터나 시스템 상태를 일관된 방식으로 처리할 수 있게 해준다.