MANIFEST.in
setup.py
labml_nn/__init__.py
labml_nn.egg-info/PKG-INFO
labml_nn.egg-info/SOURCES.txt
labml_nn.egg-info/dependency_links.txt
labml_nn.egg-info/requires.txt
labml_nn.egg-info/top_level.txt
labml_nn/activations/__init__.py
labml_nn/activations/swish.py
labml_nn/activations/fta/__init__.py
labml_nn/activations/fta/experiment.py
labml_nn/adaptive_computation/__init__.py
labml_nn/adaptive_computation/parity.py
labml_nn/adaptive_computation/ponder_net/__init__.py
labml_nn/adaptive_computation/ponder_net/experiment.py
labml_nn/capsule_networks/__init__.py
labml_nn/capsule_networks/mnist.py
labml_nn/cfr/__init__.py
labml_nn/cfr/analytics.py
labml_nn/cfr/infoset_saver.py
labml_nn/cfr/kuhn/__init__.py
labml_nn/conv_mixer/__init__.py
labml_nn/conv_mixer/experiment.py
labml_nn/diffusion/__init__.py
labml_nn/diffusion/ddpm/__init__.py
labml_nn/diffusion/ddpm/evaluate.py
labml_nn/diffusion/ddpm/experiment.py
labml_nn/diffusion/ddpm/unet.py
labml_nn/diffusion/ddpm/utils.py
labml_nn/diffusion/stable_diffusion/__init__.py
labml_nn/diffusion/stable_diffusion/latent_diffusion.py
labml_nn/diffusion/stable_diffusion/util.py
labml_nn/diffusion/stable_diffusion/model/__init__.py
labml_nn/diffusion/stable_diffusion/model/autoencoder.py
labml_nn/diffusion/stable_diffusion/model/clip_embedder.py
labml_nn/diffusion/stable_diffusion/model/unet.py
labml_nn/diffusion/stable_diffusion/model/unet_attention.py
labml_nn/diffusion/stable_diffusion/sampler/__init__.py
labml_nn/diffusion/stable_diffusion/sampler/ddim.py
labml_nn/diffusion/stable_diffusion/sampler/ddpm.py
labml_nn/diffusion/stable_diffusion/scripts/__init__.py
labml_nn/diffusion/stable_diffusion/scripts/image_to_image.py
labml_nn/diffusion/stable_diffusion/scripts/in_paint.py
labml_nn/diffusion/stable_diffusion/scripts/text_to_image.py
labml_nn/distillation/__init__.py
labml_nn/distillation/large.py
labml_nn/distillation/small.py
labml_nn/experiments/__init__.py
labml_nn/experiments/arithmetic_dataset.py
labml_nn/experiments/cifar10.py
labml_nn/experiments/mnist.py
labml_nn/experiments/nlp_autoregression.py
labml_nn/experiments/nlp_classification.py
labml_nn/gan/__init__.py
labml_nn/gan/cycle_gan/__init__.py
labml_nn/gan/dcgan/__init__.py
labml_nn/gan/original/__init__.py
labml_nn/gan/original/experiment.py
labml_nn/gan/stylegan/__init__.py
labml_nn/gan/stylegan/experiment.py
labml_nn/gan/wasserstein/__init__.py
labml_nn/gan/wasserstein/experiment.py
labml_nn/gan/wasserstein/gradient_penalty/__init__.py
labml_nn/gan/wasserstein/gradient_penalty/experiment.py
labml_nn/graphs/__init__.py
labml_nn/graphs/gat/__init__.py
labml_nn/graphs/gat/experiment.py
labml_nn/graphs/gatv2/__init__.py
labml_nn/graphs/gatv2/experiment.py
labml_nn/hypernetworks/__init__.py
labml_nn/hypernetworks/experiment.py
labml_nn/hypernetworks/hyper_lstm.py
labml_nn/lstm/__init__.py
labml_nn/neox/__init__.py
labml_nn/neox/checkpoint.py
labml_nn/neox/model.py
labml_nn/neox/tokenizer.py
labml_nn/neox/evaluation/__init__.py
labml_nn/neox/evaluation/half_precision.py
labml_nn/neox/evaluation/llm_int8.py
labml_nn/neox/samples/__init__.py
labml_nn/neox/samples/finetune.py
labml_nn/neox/samples/generate.py
labml_nn/neox/samples/llm_int8.py
labml_nn/neox/utils/__init__.py
labml_nn/neox/utils/cache.py
labml_nn/neox/utils/finetune.py
labml_nn/neox/utils/llm_int8.py
labml_nn/neox/utils/text_dataset.py
labml_nn/neox/utils/trainer.py
labml_nn/normalization/__init__.py
labml_nn/normalization/batch_channel_norm/__init__.py
labml_nn/normalization/batch_norm/__init__.py
labml_nn/normalization/batch_norm/cifar10.py
labml_nn/normalization/batch_norm/mnist.py
labml_nn/normalization/deep_norm/__init__.py
labml_nn/normalization/deep_norm/experiment.py
labml_nn/normalization/group_norm/__init__.py
labml_nn/normalization/group_norm/experiment.py
labml_nn/normalization/instance_norm/__init__.py
labml_nn/normalization/instance_norm/experiment.py
labml_nn/normalization/layer_norm/__init__.py
labml_nn/normalization/weight_standardization/__init__.py
labml_nn/normalization/weight_standardization/conv2d.py
labml_nn/normalization/weight_standardization/experiment.py
labml_nn/optimizers/__init__.py
labml_nn/optimizers/ada_belief.py
labml_nn/optimizers/adam.py
labml_nn/optimizers/adam_fp16.py
labml_nn/optimizers/adam_warmup.py
labml_nn/optimizers/adam_warmup_cosine_decay.py
labml_nn/optimizers/amsgrad.py
labml_nn/optimizers/configs.py
labml_nn/optimizers/mnist_experiment.py
labml_nn/optimizers/noam.py
labml_nn/optimizers/performance_test.py
labml_nn/optimizers/radam.py
labml_nn/optimizers/sophia.py
labml_nn/recurrent_highway_networks/__init__.py
labml_nn/resnet/__init__.py
labml_nn/resnet/experiment.py
labml_nn/rl/__init__.py
labml_nn/rl/game.py
labml_nn/rl/dqn/__init__.py
labml_nn/rl/dqn/experiment.py
labml_nn/rl/dqn/model.py
labml_nn/rl/dqn/replay_buffer.py
labml_nn/rl/ppo/__init__.py
labml_nn/rl/ppo/experiment.py
labml_nn/rl/ppo/gae.py
labml_nn/sampling/__init__.py
labml_nn/sampling/experiment.py
labml_nn/sampling/experiment_tiny.py
labml_nn/sampling/greedy.py
labml_nn/sampling/nucleus.py
labml_nn/sampling/temperature.py
labml_nn/sampling/top_k.py
labml_nn/scaling/__init__.py
labml_nn/scaling/zero3/__init__.py
labml_nn/scaling/zero3/finetune_neox.py
labml_nn/sketch_rnn/__init__.py
labml_nn/transformers/__init__.py
labml_nn/transformers/configs.py
labml_nn/transformers/feed_forward.py
labml_nn/transformers/label_smoothing_loss.py
labml_nn/transformers/mha.py
labml_nn/transformers/models.py
labml_nn/transformers/positional_encoding.py
labml_nn/transformers/relative_mha.py
labml_nn/transformers/utils.py
labml_nn/transformers/aft/__init__.py
labml_nn/transformers/aft/experiment.py
labml_nn/transformers/alibi/__init__.py
labml_nn/transformers/alibi/experiment.py
labml_nn/transformers/basic/__init__.py
labml_nn/transformers/basic/autoregressive_experiment.py
labml_nn/transformers/basic/with_sophia.py
labml_nn/transformers/compressive/__init__.py
labml_nn/transformers/compressive/experiment.py
labml_nn/transformers/fast_weights/__init__.py
labml_nn/transformers/fast_weights/experiment.py
labml_nn/transformers/fast_weights/token_wise.py
labml_nn/transformers/feedback/__init__.py
labml_nn/transformers/feedback/experiment.py
labml_nn/transformers/fnet/__init__.py
labml_nn/transformers/fnet/experiment.py
labml_nn/transformers/glu_variants/__init__.py
labml_nn/transformers/glu_variants/experiment.py
labml_nn/transformers/glu_variants/simple.py
labml_nn/transformers/gmlp/__init__.py
labml_nn/transformers/gmlp/experiment.py
labml_nn/transformers/gpt/__init__.py
labml_nn/transformers/hour_glass/__init__.py
labml_nn/transformers/hour_glass/experiment.py
labml_nn/transformers/knn/__init__.py
labml_nn/transformers/knn/build_index.py
labml_nn/transformers/knn/eval_knn.py
labml_nn/transformers/knn/train_model.py
labml_nn/transformers/mlm/__init__.py
labml_nn/transformers/mlm/experiment.py
labml_nn/transformers/mlp_mixer/__init__.py
labml_nn/transformers/mlp_mixer/experiment.py
labml_nn/transformers/primer_ez/__init__.py
labml_nn/transformers/primer_ez/efficient.py
labml_nn/transformers/primer_ez/experiment.py
labml_nn/transformers/primer_ez/variations.py
labml_nn/transformers/retro/__init__.py
labml_nn/transformers/retro/bert_embeddings.py
labml_nn/transformers/retro/database.py
labml_nn/transformers/retro/dataset.py
labml_nn/transformers/retro/model.py
labml_nn/transformers/retro/train.py
labml_nn/transformers/rope/__init__.py
labml_nn/transformers/rope/experiment.py
labml_nn/transformers/rope/value_pe/__init__.py
labml_nn/transformers/rope/value_pe/arithmetic_experiment.py
labml_nn/transformers/rope/value_pe/experiment.py
labml_nn/transformers/switch/__init__.py
labml_nn/transformers/switch/experiment.py
labml_nn/transformers/vit/__init__.py
labml_nn/transformers/vit/experiment.py
labml_nn/transformers/xl/__init__.py
labml_nn/transformers/xl/experiment.py
labml_nn/transformers/xl/relative_mha.py
labml_nn/uncertainty/__init__.py
labml_nn/uncertainty/evidence/__init__.py
labml_nn/uncertainty/evidence/experiment.py
labml_nn/unet/__init__.py
labml_nn/unet/carvana.py
labml_nn/unet/experiment.py
labml_nn/utils/__init__.py
labml_nn/utils/tokenizer.py
utils/__init__.py
utils/diagrams.py
utils/papers_list.py
utils/sitemap.py