Companies that use Trax

It helps you understand and explore advanced deep learning. It is actively used and maintained in the Google Brain team. You can use It either as a library from your own python scripts and notebooks or as a binary from the shell, which can be more convenient for training large models. It includes a number of deep learning models (ResNet, Transformer, RNNs, ...) and has bindings to a large number of deep learning datasets, including Tensor2Tensor and TensorFlow datasets. It runs without any changes on CPUs, GPUs and TPUs.