Trax
latest
Introductory Notebooks
Trax Quick Intro
Trax Layers Intro
Using Trax with TensorFlow NumPy and Keras
Packages/modules
trax.*
Trax
Docs
»
Trax Tutorials
Edit on GitHub
Trax Tutorials
¶
Introductory Notebooks
Trax Quick Intro
1. Run a pre-trained Transformer
2. Features and resources
3. Walkthrough
Trax Layers Intro
1. Layers
2. Inputs and Outputs
3. Defining New Layer Classes
4. Testing and Debugging Layer Classes
Using Trax with TensorFlow NumPy and Keras
1. Trax with TensorFlow NumPy
2. Convert Trax to Keras
3. Exporting Trax Models for Deployment
Trax API
¶
Packages/modules
trax.*
fastmath.*
layers.*
models.*
data.*
optimizers.*
supervised.*
rl.*
shapes
trainer
rl_trainer
trax2keras
Indices and Tables
¶
Index
Module Index
Search Page