- main() (in module modelzoo.common.pytorch.run_gpt_inference)
- make_key_padding_mask_broadcastable() (in module modelzoo.transformers.pytorch.transformer_utils)
- make_mask_drop_minimum() (in module cerebras_pytorch.sparse.utils)
- make_mask_grow_maximum() (in module cerebras_pytorch.sparse.utils)
- make_mask_topk_sparsity() (in module cerebras_pytorch.sparse.utils)
- make_schedule() (in module cerebras_pytorch.sparse.dynamic)
- make_sparse_mask_broadcastable() (in module modelzoo.transformers.pytorch.transformer_utils)
- manage_optimizer_state_sparsity() (cerebras_pytorch.sparse.base.BaseSparsityOptimizer method)
- maskedLmInstance (class in modelzoo.transformers.data_processing.utils)
- maybe_tie_lm_head() (in module modelzoo.common.pytorch.model_utils.checkpoint_converters.helper)
- MeanIOUMetric (class in cerebras_pytorch.metrics)
- Metric (class in cerebras_pytorch.metrics)
- Metrics (class in cerebras_pytorch.utils.benchmark.utils.dataloader)
- Mixture (class in modelzoo.transformers.data_processing.h5_map_dataset.readers)
- mlm_xentropy_loss() (modelzoo.transformers.pytorch.bert.model.BertForPreTrainingModel method)
- MLMOnlyInstance (class in modelzoo.transformers.data_processing.bert.mlm_only_processor)
- MNIST (class in modelzoo.fc_mnist.pytorch.model)
- MNISTModel (class in modelzoo.fc_mnist.pytorch.model)
- MNLIDataProcessor (class in modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor)
- MNLIDataset (class in modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor)
-
modelzoo
-
modelzoo.common
-
modelzoo.common.input
-
modelzoo.common.input.utils
-
modelzoo.common.model_utils
-
modelzoo.common.model_utils.count_lines
-
modelzoo.common.pytorch
-
modelzoo.common.pytorch.dump_context
-
modelzoo.common.pytorch.half_dtype
-
modelzoo.common.pytorch.input
-
modelzoo.common.pytorch.input.utils
-
modelzoo.common.pytorch.input_utils
-
modelzoo.common.pytorch.layers
-
modelzoo.common.pytorch.model_utils
-
modelzoo.common.pytorch.model_utils.activations
-
modelzoo.common.pytorch.model_utils.BertPretrainModelLoss
-
modelzoo.common.pytorch.model_utils.checkpoint_converters
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.artifacts
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.bert
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_180b
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.helper
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.llama
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.mistral
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.mup
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints
-
modelzoo.common.pytorch.model_utils.checkpoint_converters.t5
-
modelzoo.common.pytorch.model_utils.convert_checkpoint
-
modelzoo.common.pytorch.model_utils.convert_config_to_mup
-
modelzoo.common.pytorch.model_utils.create_initializer
-
modelzoo.common.pytorch.model_utils.DPOLoss
-
modelzoo.common.pytorch.model_utils.GPTLMHeadModelLoss
-
modelzoo.common.pytorch.model_utils.norms
-
modelzoo.common.pytorch.model_utils.RotaryPositionEmbeddingHelper
-
modelzoo.common.pytorch.model_utils.T5ForConditionalGenerationLoss
-
modelzoo.common.pytorch.model_utils.vocab_utils
-
modelzoo.common.pytorch.model_utils.weight_initializers
-
modelzoo.common.pytorch.run_cstorch_eval_harness
-
modelzoo.common.pytorch.run_cstorch_flow
-
modelzoo.common.pytorch.run_gpt_inference
-
modelzoo.common.pytorch.run_utils
-
modelzoo.common.pytorch.utils
-
modelzoo.common.run_utils
-
modelzoo.common.run_utils.cli_parser
-
modelzoo.common.run_utils.cli_pytorch
-
modelzoo.common.run_utils.utils
-
modelzoo.fc_mnist
-
modelzoo.fc_mnist.pytorch
-
modelzoo.fc_mnist.pytorch.data
-
modelzoo.fc_mnist.pytorch.model
-
modelzoo.fc_mnist.pytorch.prepare_data
-
modelzoo.fc_mnist.pytorch.run
-
modelzoo.fc_mnist.pytorch.utils
-
modelzoo.transformers
-
modelzoo.transformers.data_processing
-
modelzoo.transformers.data_processing.bert
-
modelzoo.transformers.data_processing.bert.bertsum_data_processor
-
modelzoo.transformers.data_processing.bert.dynamic_processor
-
modelzoo.transformers.data_processing.bert.mlm_only_processor
-
modelzoo.transformers.data_processing.bert.ner_data_processor
-
modelzoo.transformers.data_processing.bert.sentence_pair_processor
-
modelzoo.transformers.data_processing.GenericDataProcessor
-
modelzoo.transformers.data_processing.h5_map_dataset
-
modelzoo.transformers.data_processing.h5_map_dataset.dataset
-
modelzoo.transformers.data_processing.h5_map_dataset.preprocess_pile
-
modelzoo.transformers.data_processing.h5_map_dataset.readers
-
modelzoo.transformers.data_processing.h5_map_dataset.samplers
-
modelzoo.transformers.data_processing.HDF5IterableDataProcessor
-
modelzoo.transformers.data_processing.HDF5IterableDataset
-
modelzoo.transformers.data_processing.huggingface
-
modelzoo.transformers.data_processing.huggingface.CSDataCollatorForLanguageModeling
-
modelzoo.transformers.data_processing.huggingface.HF_converter_example_BookCorpus
-
modelzoo.transformers.data_processing.huggingface.HF_converter_example_Eli5
-
modelzoo.transformers.data_processing.huggingface.HuggingFace_BookCorpus
-
modelzoo.transformers.data_processing.huggingface.HuggingFace_Eli5
-
modelzoo.transformers.data_processing.huggingface.HuggingFaceDataProcessor
-
modelzoo.transformers.data_processing.qa
-
modelzoo.transformers.data_processing.qa.qa_utils
-
modelzoo.transformers.data_processing.qa.write_csv_qa
-
modelzoo.transformers.data_processing.scripts
-
modelzoo.transformers.data_processing.scripts.chunk_preprocessing
-
modelzoo.transformers.data_processing.scripts.chunk_preprocessing.chunk_data_preprocessor
-
modelzoo.transformers.data_processing.scripts.chunk_preprocessing.create_hdf5_dataset
-
modelzoo.transformers.data_processing.scripts.chunk_preprocessing.data_reader
-
modelzoo.transformers.data_processing.scripts.chunk_preprocessing.fim_data_token_generator
-
modelzoo.transformers.data_processing.scripts.chunk_preprocessing.lm_data_token_generator
-
modelzoo.transformers.data_processing.scripts.chunk_preprocessing.summarization_data_token_generator
-
modelzoo.transformers.data_processing.scripts.chunk_preprocessing.utils
-
modelzoo.transformers.data_processing.scripts.hdf5_preprocessing
-
modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.convert_dataset_to_HDF5
-
modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.create_hdf5_dataset
-
modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor
-
modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor
-
modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors
-
modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor
-
modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils
-
modelzoo.transformers.data_processing.scripts.hdf5_shuffling
-
modelzoo.transformers.data_processing.scripts.hdf5_shuffling.h5_dataset_shuffle
-
modelzoo.transformers.data_processing.scripts.pile
-
modelzoo.transformers.data_processing.scripts.pile.download
-
modelzoo.transformers.data_processing.scripts.pubmed
-
modelzoo.transformers.data_processing.scripts.pubmed.preprocess
-
modelzoo.transformers.data_processing.scripts.pubmed.preprocess.Downloader
-
modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextFormatting
-
modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextSharding
-
modelzoo.transformers.data_processing.slimpajama
-
modelzoo.transformers.data_processing.slimpajama.dedup
-
modelzoo.transformers.data_processing.slimpajama.dedup.dedup_train
-
modelzoo.transformers.data_processing.slimpajama.dedup.generate_connected_components
-
modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicate_pairs
-
modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicates_dict
-
modelzoo.transformers.data_processing.slimpajama.dedup.to_hash
-
modelzoo.transformers.data_processing.slimpajama.main
-
modelzoo.transformers.data_processing.slimpajama.preprocessing
-
modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets
-
modelzoo.transformers.data_processing.slimpajama.preprocessing.filter
-
modelzoo.transformers.data_processing.slimpajama.preprocessing.normalize_text
-
modelzoo.transformers.data_processing.slimpajama.preprocessing.shuffle_holdout
-
modelzoo.transformers.data_processing.slimpajama.split_dataset
-
modelzoo.transformers.data_processing.slimpajama.utils
-
modelzoo.transformers.data_processing.slimpajama.utils.utils
-
modelzoo.transformers.data_processing.tokenizers
-
modelzoo.transformers.data_processing.tokenizers.BPETokenizer
-
modelzoo.transformers.data_processing.tokenizers.HFTokenizer
-
modelzoo.transformers.data_processing.tokenizers.Tokenization
-
modelzoo.transformers.data_processing.utils
-
modelzoo.transformers.pytorch
-
modelzoo.transformers.pytorch.bert
-
modelzoo.transformers.pytorch.bert.bert_finetune_models
-
modelzoo.transformers.pytorch.bert.bert_model
-
modelzoo.transformers.pytorch.bert.bert_pretrain_models
-
modelzoo.transformers.pytorch.bert.data
-
modelzoo.transformers.pytorch.bert.fine_tuning
-
modelzoo.transformers.pytorch.bert.fine_tuning.classifier
-
modelzoo.transformers.pytorch.bert.fine_tuning.classifier.data
-
modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input
-
modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor
-
modelzoo.transformers.pytorch.bert.fine_tuning.classifier.model
-
modelzoo.transformers.pytorch.bert.fine_tuning.classifier.run
-
modelzoo.transformers.pytorch.bert.fine_tuning.classifier.utils
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.data
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.write_csv_data
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.model
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.run
-
modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.utils
-
modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier
-
modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.data
-
modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input
-
modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor
-
modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner
-
modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.model
-
modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.run
-
modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.utils
-
modelzoo.transformers.pytorch.bert.input
-
modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor
-
modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor
-
modelzoo.transformers.pytorch.bert.input.scripts
-
modelzoo.transformers.pytorch.bert.input.scripts.create_csv
-
modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only
-
modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only_static_masking
-
modelzoo.transformers.pytorch.bert.input.scripts.create_csv_static_masking
-
modelzoo.transformers.pytorch.bert.input.scripts.create_hdf5_files
-
modelzoo.transformers.pytorch.bert.input.scripts.parser_utils
-
modelzoo.transformers.pytorch.bert.input.utils
-
modelzoo.transformers.pytorch.bert.model
-
modelzoo.transformers.pytorch.bert.run
-
modelzoo.transformers.pytorch.bert.utils
-
modelzoo.transformers.pytorch.bloom
-
modelzoo.transformers.pytorch.bloom.run
-
modelzoo.transformers.pytorch.btlm
-
modelzoo.transformers.pytorch.btlm.run
-
modelzoo.transformers.pytorch.falcon
-
modelzoo.transformers.pytorch.falcon.run
-
modelzoo.transformers.pytorch.gpt2
-
modelzoo.transformers.pytorch.gpt2.data
-
modelzoo.transformers.pytorch.gpt2.gpt2_model
-
modelzoo.transformers.pytorch.gpt2.input
-
modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor
-
modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor
-
modelzoo.transformers.pytorch.gpt2.input.GptHDF5DataProcessor
-
modelzoo.transformers.pytorch.gpt2.input.GptHDF5MapDataProcessor
-
modelzoo.transformers.pytorch.gpt2.input.GptTextDataProcessor
-
modelzoo.transformers.pytorch.gpt2.input.HuggingFaceDataProcessorEli5
-
modelzoo.transformers.pytorch.gpt2.input.HuggingFaceIterableDataProcessorEli5
-
modelzoo.transformers.pytorch.gpt2.input.InferenceDataProcessor
-
modelzoo.transformers.pytorch.gpt2.input.scripts
-
modelzoo.transformers.pytorch.gpt2.input.scripts.data_processor_utils
-
modelzoo.transformers.pytorch.gpt2.model
-
modelzoo.transformers.pytorch.gpt2.run
-
modelzoo.transformers.pytorch.gpt2.scripts
-
modelzoo.transformers.pytorch.gpt2.sparse_mask
-
modelzoo.transformers.pytorch.gpt2.utils
-
modelzoo.transformers.pytorch.gpt3
-
modelzoo.transformers.pytorch.gpt3.run
-
modelzoo.transformers.pytorch.gptj
-
modelzoo.transformers.pytorch.gptj.fine_tuning
-
modelzoo.transformers.pytorch.gptj.fine_tuning.continuous_pretraining
-
modelzoo.transformers.pytorch.gptj.fine_tuning.continuous_pretraining.data_processing
-
modelzoo.transformers.pytorch.gptj.fine_tuning.continuous_pretraining.data_processing.split_trc_dataset
-
modelzoo.transformers.pytorch.gptj.gptj_model
-
modelzoo.transformers.pytorch.gptj.model
-
modelzoo.transformers.pytorch.gptj.run
-
modelzoo.transformers.pytorch.gptj.utils
-
modelzoo.transformers.pytorch.input_utils
-
modelzoo.transformers.pytorch.layers_api_demo
|
-
modelzoo.transformers.pytorch.layers_api_demo.cb_transformer
-
modelzoo.transformers.pytorch.layers_api_demo.data
-
modelzoo.transformers.pytorch.layers_api_demo.model
-
modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer
-
modelzoo.transformers.pytorch.layers_api_demo.run
-
modelzoo.transformers.pytorch.llama
-
modelzoo.transformers.pytorch.llama.run
-
modelzoo.transformers.pytorch.mistral
-
modelzoo.transformers.pytorch.mistral.run
-
modelzoo.transformers.pytorch.mpt
-
modelzoo.transformers.pytorch.mpt.run
-
modelzoo.transformers.pytorch.santacoder
-
modelzoo.transformers.pytorch.santacoder.run
-
modelzoo.transformers.pytorch.starcoder
-
modelzoo.transformers.pytorch.starcoder.run
-
modelzoo.transformers.pytorch.t5
-
modelzoo.transformers.pytorch.t5.data
-
modelzoo.transformers.pytorch.t5.input
-
modelzoo.transformers.pytorch.t5.input.data_processing
-
modelzoo.transformers.pytorch.t5.input.data_processing.preprocess_t5_dataset
-
modelzoo.transformers.pytorch.t5.input.data_processing.test_convert_t5_dataset
-
modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor
-
modelzoo.transformers.pytorch.t5.input.T5HDF5DataProcessor
-
modelzoo.transformers.pytorch.t5.input.utils
-
modelzoo.transformers.pytorch.t5.model
-
modelzoo.transformers.pytorch.t5.run
-
modelzoo.transformers.pytorch.t5.t5_model
-
modelzoo.transformers.pytorch.t5.utils
-
modelzoo.transformers.pytorch.transformer
-
modelzoo.transformers.pytorch.transformer.data
-
modelzoo.transformers.pytorch.transformer.input
-
modelzoo.transformers.pytorch.transformer.input.data_processing
-
modelzoo.transformers.pytorch.transformer.input.data_processing.create_meta
-
modelzoo.transformers.pytorch.transformer.input.data_processing.split_files
-
modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor
-
modelzoo.transformers.pytorch.transformer.run
-
modelzoo.transformers.pytorch.transformer.utils
-
modelzoo.transformers.pytorch.transformer_utils
-
modelzoo.transformers.utils
-
modelzoo.transformers.utils.attention_utils
-
modelzoo.vision
-
modelzoo.vision.pytorch
-
modelzoo.vision.pytorch.dit
-
modelzoo.vision.pytorch.dit.checkpoint_converter
-
modelzoo.vision.pytorch.dit.checkpoint_converter.vae_hf_cs
-
modelzoo.vision.pytorch.dit.data
-
modelzoo.vision.pytorch.dit.display_images
-
modelzoo.vision.pytorch.dit.input
-
modelzoo.vision.pytorch.dit.input.DiffusionBaseProcessor
-
modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor
-
modelzoo.vision.pytorch.dit.input.scripts
-
modelzoo.vision.pytorch.dit.input.scripts.create_imagenet_latents
-
modelzoo.vision.pytorch.dit.input.transforms
-
modelzoo.vision.pytorch.dit.layers
-
modelzoo.vision.pytorch.dit.layers.DiTDecoder
-
modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer
-
modelzoo.vision.pytorch.dit.layers.GaussianDiffusion
-
modelzoo.vision.pytorch.dit.layers.RegressionHead
-
modelzoo.vision.pytorch.dit.layers.schedulers
-
modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer
-
modelzoo.vision.pytorch.dit.layers.vae
-
modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D
-
modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D
-
modelzoo.vision.pytorch.dit.layers.vae.UNetMidBlock2D
-
modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D
-
modelzoo.vision.pytorch.dit.layers.vae.utils
-
modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder
-
modelzoo.vision.pytorch.dit.layers.vae.VAEEncoder
-
modelzoo.vision.pytorch.dit.layers.vae.VAEModel
-
modelzoo.vision.pytorch.dit.model
-
modelzoo.vision.pytorch.dit.modeling_dit
-
modelzoo.vision.pytorch.dit.pipeline
-
modelzoo.vision.pytorch.dit.run
-
modelzoo.vision.pytorch.dit.sample_generator
-
modelzoo.vision.pytorch.dit.sample_generator_dit
-
modelzoo.vision.pytorch.dit.samplers
-
modelzoo.vision.pytorch.dit.samplers.DDIMSampler
-
modelzoo.vision.pytorch.dit.samplers.DDPMSampler
-
modelzoo.vision.pytorch.dit.samplers.get_sampler
-
modelzoo.vision.pytorch.dit.samplers.sampler_utils
-
modelzoo.vision.pytorch.dit.samplers.SamplerBase
-
modelzoo.vision.pytorch.dit.utils
-
modelzoo.vision.pytorch.input
-
modelzoo.vision.pytorch.input.classification
-
modelzoo.vision.pytorch.input.classification.mixup
-
modelzoo.vision.pytorch.input.classification.utils
-
modelzoo.vision.pytorch.input.preprocessing
-
modelzoo.vision.pytorch.input.transforms
-
modelzoo.vision.pytorch.input.utils
-
modelzoo.vision.pytorch.utils
-
modelzoo.vision.pytorch.utils.run_utils
-
module
- MultiheadAttention (class in modelzoo.common.pytorch.layers)
- MultiplicativeLR (class in cerebras_pytorch.optim.lr_scheduler)
- MultiQueryAttention (class in modelzoo.common.pytorch.layers)
- MultiStepLR (class in cerebras_pytorch.optim.lr_scheduler)
|