Python Module Index

t


	t
	`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`