MANIFEST.in
README.md
setup.py
version.txt
requirements/requirements-test.txt
requirements/requirements.txt
tests/utils/__init__.py
tests/utils/dist_test.py
titans/__init__.py
titans.egg-info/PKG-INFO
titans.egg-info/SOURCES.txt
titans.egg-info/dependency_links.txt
titans.egg-info/requires.txt
titans.egg-info/top_level.txt
titans/dataloader/__init__.py
titans/dataloader/bert/__init__.py
titans/dataloader/bert/bert_pretrain.py
titans/dataloader/bert/parquet_dataset.py
titans/dataloader/cifar10/__init__.py
titans/dataloader/cifar10/torchvision_cifar10.py
titans/dataloader/imagenet/__init__.py
titans/dataloader/imagenet/dali_imagenet.py
titans/dataloader/imagenet/torchvision_imagenet.py
titans/dataloader/utils/__init__.py
titans/dataloader/utils/rand_augment.py
titans/decorator/__init__.py
titans/decorator/no_support.py
titans/layer/__init__.py
titans/layer/init_rules.py
titans/layer/attention/__init__.py
titans/layer/attention/detr_attention.py
titans/layer/attention/gpt_attention.py
titans/layer/attention/transformer_attention.py
titans/layer/attention/vit_attention.py
titans/layer/attention/vit_moe_attention.py
titans/layer/batchnorm/__init__.py
titans/layer/batchnorm/frozen_batchnorm_2d.py
titans/layer/block/__init__.py
titans/layer/block/deepnet_block.py
titans/layer/block/detr_block.py
titans/layer/block/gpt_block.py
titans/layer/block/transformer_decoder.py
titans/layer/block/transformer_encoder.py
titans/layer/block/utils.py
titans/layer/block/vit_block.py
titans/layer/embedding/__init__.py
titans/layer/embedding/gpt_embedding.py
titans/layer/embedding/vit_embedding.py
titans/layer/head/__init__.py
titans/layer/head/gpt_lm_head.py
titans/layer/head/vit_head.py
titans/layer/mlp/__init__.py
titans/layer/mlp/detr_mlp.py
titans/layer/mlp/transformer_mlp.py
titans/layer/mlp/vit_mlp.py
titans/layer/mlp/vit_moe_mlp.py
titans/loss/__init__.py
titans/loss/embedding_loss/__init__.py
titans/loss/embedding_loss/embedding_loss.py
titans/loss/lm_loss/__init__.py
titans/loss/lm_loss/gpt_lmloss.py
titans/loss/mlm_loss/__init__.py
titans/loss/mlm_loss/mlm_loss.py
titans/loss/vocab_cross_entropy/__init__.py
titans/loss/vocab_cross_entropy/vocab_cross_entropy.py
titans/model/__init__.py
titans/model/helper.py
titans/model/deepnet/__init__.py
titans/model/deepnet/deepnet.py
titans/model/detr/__init__.py
titans/model/detr/detr.py
titans/model/gpt/__init__.py
titans/model/gpt/gpt.py
titans/model/knowledge_graph_embedding/__init__.py
titans/model/knowledge_graph_embedding/knowledge_graph_embedding.py
titans/model/knowledge_graph_embedding/dataloader/__init__.py
titans/model/knowledge_graph_embedding/dataloader/dataloader.py
titans/model/moe/__init__.py
titans/model/moe/gpt_moe.py
titans/model/moe/util.py
titans/model/moe/vit_moe.py
titans/model/moe/widenet.py
titans/model/transformer/__init__.py
titans/model/transformer/transformer.py
titans/model/vilt/__init__.py
titans/model/vilt/vilt.py
titans/model/vit/__init__.py
titans/model/vit/vit.py
titans/utils/__init__.py
titans/utils/context.py
titans/utils/tensor_parallel_data_split.py
titans/utils/utils.py