flax on Hacker News: 1 stories with screenshots

Kauldron: Modular, scalable library to train ML models (github.com)

20 points by nateb2022 4 days ago | 2 comments

GitHub repository for Kauldron, Google Research's modular JAX/Flax library for training machine learning models with a focus on research velocity.