Introductory Notebooks
Packages/modules
trax
trax.data.inputs
trax.data.tf_inputs
trax.fastmath.ops
trax.layers.acceleration
trax.layers.activation_fns
trax.layers.attention
trax.layers.base
trax.layers.combinators
trax.layers.convolution
trax.layers.core
trax.layers.initializers
trax.layers.metrics
trax.layers.normalization
trax.layers.pooling
trax.layers.research.efficient_attention
trax.layers.research.position_encodings
trax.layers.reversible
trax.layers.rnn
trax.models.atari_cnn
trax.models.mlp
trax.models.neural_gpu
trax.models.reformer.reformer
trax.models.research.bert
trax.models.resnet
trax.models.rl
trax.models.rnn
trax.models.transformer
trax.optimizers.adafactor
trax.optimizers.adam
trax.optimizers.base
trax.optimizers.momentum
trax.optimizers.rms_prop
trax.optimizers.sm3
trax.rl.actor_critic
trax.rl.actor_critic_joint
trax.rl.advantages
trax.rl.distributions
trax.rl.normalization
trax.rl.rl_layers
trax.rl.serialization_utils
trax.rl.space_serializer
trax.rl.task
trax.rl.training
trax.rl_trainer
trax.shapes
trax.supervised.decoding
trax.supervised.lr_schedules
trax.supervised.training
trax.trainer
trax.trax2keras