umangjpatel / keraxLinks
Keras-like APIs for JAX framework
☆49Updated 2 years ago
Alternatives and similar repositories for kerax
Users that are interested in kerax are comparing it to the libraries listed below
Sorting:
- A queue service for quickly developing scripts that use all your GPUs efficiently☆85Updated 2 years ago
- A Pytree Module system for Deep Learning in JAX