Google JAX, Flax, Optax

leehyuna·2023년 5월 31일


목록 보기


JAX is Autograd and XLA, brought together for high-performance machine learning research.
JAX is NumPy on the CPU, GPU, and TPU, with great automatic differentiation for high-performance machine learning research.

🔎 Flax

Flax is a neural network library and ecosystem for JAX designed for flexibility.
Flax delivers an end-to-end and flexible user experience for researchers who use JAX with neural networks.

🔎 Optax

Optax is a gradient processing and optimization library for JAX.
Optax is designed to facilitate research by providing building blocks that can be easily recombined in custom ways.

AI, Computer Vision, HCI

0개의 댓글