Article
Easy JAX training loops with Flax and Optax
By
•
•
6Naufal Hafish Junianto
Nongpal
AI & ML interests
Computer Vision, Natural Language Processing, and Reinforcement Learning.
Recent Activity
upvoted
an
article
about 2 months ago
Easy JAX training loops with Flax and Optax
liked
a dataset
about 2 months ago
cognitivecomputations/samantha-data
Organizations
None yet
Nongpal's activity
upvoted
an
article
about 2 months ago
upvoted
a
collection
6 months ago