search

caelum02 / Lux-AI-Season-2

Lux AI Season 2 - NeurIPS Stage | Team Martian
1 stars 0 forks source link

Jax deep learning tutorial #1

Open caelum02 opened 9 months ago

caelum02 commented 9 months ago

University of Virginia - Deep Learning (Jax+Flax)

Available Package 에 포함된 것 가운데 JAX와 함께 유용하게 사용되는 라이브러리

Flax: A neural network library and ecosystem for JAX designed for flexibility

Optax: gradient processing and optimization library for JAX

gyusang commented 9 months ago

아 Flax 쓰는 거 같더라 맞아

caelum02 commented 9 months ago

추가 완료