하이크
JAX 위에서 동작하는 객체 지향 신경망 라이브러리로, DeepMind에서 개발했다. Sonnet과 유사한 프로그래밍 모델을 제공하면서도 JAX의 함수형 프로그래밍 특성과 호환되어 복잡한 모델 구조를 간결하게 정의할 수 있다.