4 ms·
From the README: "Trax code is structured in a way that allows you to understand deep learning from scratch. We start with basic maths and go through layers, m
by ma2rten 7y ago
From the README:
"Trax code is structured in a way that allows you to understand deep learning from scratch. We start with basic maths and go through layers, models, supervised and reinforcement learning."
- zwaps 7y agoBut what is it? It says it is an API or module? Is it a DL framework? Is it an extension of TF/Jax with new models? Is it a set of tutorials? Are we to work through the code? Is a part of the doc for Jax? Is it a set of implementations of Jax code into concrete models? All these things come to mind when going through this Github. By now I think to have figured out that it is a module that implements some DL models using Jax. So it is like an extension to Jax. It "helps" you to understand how to build the models in Jax if you go through the code. You could also load these abstractions as a Python module and use them in code, but I would doubt that this would "help you understand deep learning". So the real intention is to go through the code and see how Jax implements, for example, a transformer. Alternatively, this could be an early stage of some PyTorch-type framework over Jax and TF. So essentially, another Keras.