Index

_

__add__() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures method) (modelzoo.transformers.data_processing.utils.maskedLmInstance method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction.RawResult method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner.InputFeatures method) __call__() (modelzoo.common.pytorch.dump_context.DumpContext method) (modelzoo.common.pytorch.model_utils.BertPretrainModelLoss.BertPretrainModelLoss method) (modelzoo.common.pytorch.model_utils.GPTLMHeadModelLoss.GPTLMHeadModelLoss method) (modelzoo.common.pytorch.model_utils.T5ForConditionalGenerationLoss.T5ForConditionalGenerationLoss method) (modelzoo.common.pytorch.utils.BufferedShuffleDataset method) (modelzoo.common.pytorch.utils.IterableDatasetSampler method) (modelzoo.fc_mnist.pytorch.model.MNIST method) (modelzoo.fc_mnist.pytorch.model.MNISTModel method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.HDF5Dataset method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.RestartableDataLoader method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.BaseSampler method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.BatchSampler method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.CBSampler method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.Sharder method) (modelzoo.transformers.data_processing.HDF5IterableDataset.HDF5IterableDataset method) (modelzoo.transformers.data_processing.HDF5IterableDataset.RestartableDataLoader method) (modelzoo.transformers.data_processing.huggingface.CSDataCollatorForLanguageModeling.CSDataCollatorForLanguageModeling method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForQuestionAnswering method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForQuestionAnsweringLoss method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSequenceClassification method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSequenceClassificationLoss method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSummarization method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSummarizationLoss method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForTokenClassification method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForTokenClassificationLoss method) (modelzoo.transformers.pytorch.bert.bert_model.BertModel method) (modelzoo.transformers.pytorch.bert.bert_model.BertPooler method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertClassifierHead method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertMLMHead method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertMLMHeadTransform method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertPretrainModel method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.ClassifierDataset method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.MNLIDataset method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.SST2Dataset method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.model.BertForSequenceClassificationModel method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor.BertSumCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.model.BertSummarizationModel method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor.BertQADataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.model.BertForQuestionAnsweringModel method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor.BertTokenClassifierDataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.model.BertForTokenClassificationModel method) (modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor.BertCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor.BertCSVDynamicMaskDataProcessor method) (modelzoo.transformers.pytorch.bert.model.BertForPreTrainingModel method) (modelzoo.transformers.pytorch.gpt2.gpt2_model.GPT2LMHeadModel method) (modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor.DummyDataset method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyIterableDataset method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyTinyIterableDataset method) (modelzoo.transformers.pytorch.gpt2.input.GptTextDataProcessor.GptTextDataProcessor method) (modelzoo.transformers.pytorch.gpt2.model.Gpt2Model method) (modelzoo.transformers.pytorch.gptj.gptj_model.GPTJModel method) (modelzoo.transformers.pytorch.gptj.model.GptjModel method) (modelzoo.transformers.pytorch.input_utils.ShardedSampler method) (modelzoo.transformers.pytorch.input_utils.SubsetSequentialSampler method) (modelzoo.transformers.pytorch.layers_api_demo.cb_transformer.TransformerModel method) (modelzoo.transformers.pytorch.layers_api_demo.data.AlphabetDataset method) (modelzoo.transformers.pytorch.layers_api_demo.model.TransformerBaseModel method) (modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.PositionalEncoding method) (modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.TransformerModel method) (modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor.T5DynamicDataProcessor method) (modelzoo.transformers.pytorch.t5.model.T5ForConditionalGenerationModel method) (modelzoo.transformers.pytorch.t5.t5_model.T5ForConditionalGeneration method) (modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor.TransformerDynamicDataProcessor method) (modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor.CategoricalDataset method) (modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor.ImageNetLatentDataset method) (modelzoo.vision.pytorch.dit.input.transforms.LabelDropout method) (modelzoo.vision.pytorch.dit.input.transforms.NoiseGenerator method) (modelzoo.vision.pytorch.dit.layers.DiTDecoder.DiTDecoder method) (modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer.DiTDecoderLayer method) (modelzoo.vision.pytorch.dit.layers.GaussianDiffusion.GaussianDiffusion method) (modelzoo.vision.pytorch.dit.layers.RegressionHead.RegressionHead method) (modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer.TimestepEmbeddingLayer method) (modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D.DownEncoderBlock2D method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Downsample2D method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.ResnetBlock2D method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Upsample2D method) (modelzoo.vision.pytorch.dit.layers.vae.UNetMidBlock2D.UNetMidBlock2D method) (modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D.UpDecoderBlock2D method) (modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder.Decoder method) (modelzoo.vision.pytorch.dit.layers.vae.VAEEncoder.Encoder method) (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) (modelzoo.vision.pytorch.dit.model.DiTModel method) (modelzoo.vision.pytorch.dit.modeling_dit.DiT method) (modelzoo.vision.pytorch.dit.pipeline.DiffusionPipeline method) (modelzoo.vision.pytorch.input.classification.dataset_factory.VisionSubset method) (modelzoo.vision.pytorch.input.classification.mixup.RandomCutmix method) (modelzoo.vision.pytorch.input.classification.mixup.RandomMixup method) (modelzoo.vision.pytorch.input.classification.sampler.RepeatedAugSampler method) (modelzoo.vision.pytorch.input.utils.FastDataLoader method) (modelzoo.vision.pytorch.input.utils.ShardedSampler method) (modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor.Cityscapes method) (modelzoo.vision.pytorch.unet.input.Hdf5BaseDataProcessor.Hdf5BaseDataProcessor method) (modelzoo.vision.pytorch.unet.input.Hdf5BaseIterDataProcessor.Hdf5BaseIterDataProcessor method) (modelzoo.vision.pytorch.unet.input.Hdf5DataProcessor.Hdf5DataProcessor method) (modelzoo.vision.pytorch.unet.input.InriaAerialDataProcessor.InriaAerialDataset method) (modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataset method) (modelzoo.vision.pytorch.unet.input.SkmDataProcessor.SkmDataProcessor method) (modelzoo.vision.pytorch.unet.layers.Decoder.Decoder method) (modelzoo.vision.pytorch.unet.layers.Encoder.Encoder method) (modelzoo.vision.pytorch.unet.layers.UNetBlock.UNetBlock method) (modelzoo.vision.pytorch.unet.model.UNetModel method) (modelzoo.vision.pytorch.unet.modeling_unet.UNet method) __init__() (cerebras_pytorch.amp.GradScaler method) (cerebras_pytorch.metrics.DiceCoefficientMetric method) (cerebras_pytorch.metrics.FBetaScoreMetric method) (cerebras_pytorch.metrics.MeanIOUMetric method) (cerebras_pytorch.metrics.Metric method) (cerebras_pytorch.optim.Adadelta method) (cerebras_pytorch.optim.Adafactor method) (cerebras_pytorch.optim.Adagrad method) (cerebras_pytorch.optim.Adam method) (cerebras_pytorch.optim.Adamax method) (cerebras_pytorch.optim.AdamW method) (cerebras_pytorch.optim.ASGD method) (cerebras_pytorch.optim.Lamb method) (cerebras_pytorch.optim.Lion method) (cerebras_pytorch.optim.lr_scheduler.ChainedScheduler method) (cerebras_pytorch.optim.lr_scheduler.ConstantLR method) (cerebras_pytorch.optim.lr_scheduler.CosineAnnealingLR method) (cerebras_pytorch.optim.lr_scheduler.CosineAnnealingWarmRestarts method) (cerebras_pytorch.optim.lr_scheduler.CosineDecayLR method) (cerebras_pytorch.optim.lr_scheduler.CyclicLR method) (cerebras_pytorch.optim.lr_scheduler.ExponentialLR method) (cerebras_pytorch.optim.lr_scheduler.InverseExponentialTimeDecayLR method) (cerebras_pytorch.optim.lr_scheduler.InverseSquareRootDecayLR method) (cerebras_pytorch.optim.lr_scheduler.LambdaLR method) (cerebras_pytorch.optim.lr_scheduler.LinearLR method) (cerebras_pytorch.optim.lr_scheduler.LRScheduler method) (cerebras_pytorch.optim.lr_scheduler.MultiplicativeLR method) (cerebras_pytorch.optim.lr_scheduler.MultiStepLR method) (cerebras_pytorch.optim.lr_scheduler.OneCycleLR method) (cerebras_pytorch.optim.lr_scheduler.PiecewiseConstantLR method) (cerebras_pytorch.optim.lr_scheduler.PolynomialLR method) (cerebras_pytorch.optim.lr_scheduler.SequentialLR method) (cerebras_pytorch.optim.lr_scheduler.StepLR method) (cerebras_pytorch.optim.NAdam method) (cerebras_pytorch.optim.Optimizer method) (cerebras_pytorch.optim.RAdam method) (cerebras_pytorch.optim.RMSprop method) (cerebras_pytorch.optim.Rprop method) (cerebras_pytorch.optim.SGD method) (cerebras_pytorch.sparse.base.BaseSparsityOptimizer method) (cerebras_pytorch.sparse.dynamic.DynamicSparsityOptimizer method) (cerebras_pytorch.sparse.dynamic.FreqSchedule method) (cerebras_pytorch.sparse.dynamic.ListSchedule method) (cerebras_pytorch.sparse.GMPSparsityOptimizer method) (cerebras_pytorch.sparse.RigLSparsityOptimizer method) (cerebras_pytorch.sparse.SETSparsityOptimizer method) (cerebras_pytorch.sparse.SparsityWrapperOptimizer method) (cerebras_pytorch.sparse.StaticSparsityOptimizer method) (cerebras_pytorch.sparse.utils.ConstantHyperParameter method) (cerebras_pytorch.sparse.utils.CosineHyperParameter method) (cerebras_pytorch.sparse.utils.CyclingHyperParameter method) (cerebras_pytorch.sparse.utils.ExpHyperParameter method) (cerebras_pytorch.sparse.utils.FirstScoreTiebreaker method) (cerebras_pytorch.sparse.utils.InputGroupScoreShaper method) (cerebras_pytorch.sparse.utils.LambdaHyperParameter method) (cerebras_pytorch.sparse.utils.LinearHyperParameter method) (cerebras_pytorch.sparse.utils.OutputGroupScoreShaper method) (cerebras_pytorch.sparse.utils.PowerHyperParameter method) (cerebras_pytorch.sparse.utils.RandomScoreTiebreaker method) (cerebras_pytorch.utils.data.DataExecutor method) (cerebras_pytorch.utils.data.DataLoader method) (cerebras_pytorch.utils.data.RestartableDataLoader method) (cerebras_pytorch.utils.data.SyntheticDataset method) (cerebras_pytorch.utils.tensorboard.SummaryReader method) (cerebras_pytorch.utils.tensorboard.SummaryWriter method) (modelzoo.common.pytorch.dump_context.DumpContext method) (modelzoo.common.pytorch.half_dtype.HalfDType method) (modelzoo.common.pytorch.layers.AdaLayerNorm method) (modelzoo.common.pytorch.layers.AlibiPositionEmbeddingLayer method) (modelzoo.common.pytorch.layers.BatchChannelNorm2D method) (modelzoo.common.pytorch.layers.BiaslessLayerNorm method) (modelzoo.common.pytorch.layers.EmbeddingLayer method) (modelzoo.common.pytorch.layers.FeedForwardNetwork method) (modelzoo.common.pytorch.layers.GPTJDecoderLayer method) (modelzoo.common.pytorch.layers.GroupInstanceNorm method) (modelzoo.common.pytorch.layers.MultiheadAttention method) (modelzoo.common.pytorch.layers.MultiQueryAttention method) (modelzoo.common.pytorch.layers.RelativePositionEmbeddingLayer method) (modelzoo.common.pytorch.layers.RMSNorm method) (modelzoo.common.pytorch.layers.Transformer method) (modelzoo.common.pytorch.layers.TransformerDecoder method) (modelzoo.common.pytorch.layers.TransformerDecoderLayer method) (modelzoo.common.pytorch.layers.TransformerEncoder method) (modelzoo.common.pytorch.layers.TransformerEncoderLayer method) (modelzoo.common.pytorch.layers.ViTEmbeddingLayer method) (modelzoo.common.pytorch.model_utils.BertPretrainModelLoss.BertPretrainModelLoss method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseDictionaryConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.ConfigConversionError method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.ConversionRule method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.EquivalentSubkey method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.ConfigConverter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.ConfigConverter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.ConfigConverter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.ConfigConverter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_Headless_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LLaMa_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LLaMa_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LlamaModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.ConfigConverter_MPT_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.ConfigConverter_Roberta_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.ConfigConverter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.ConfigConverter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints.StreamingCSWriter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints.StreamingCSWriterView method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints.StreamingShardedHFReader method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints.StreamingShardedHFWriter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.ConfigConverter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.ConfigConverter_ViTMAE_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.convert_checkpoint.CheckpointConverterCLI method) (modelzoo.common.pytorch.model_utils.GPTLMHeadModelLoss.GPTLMHeadModelLoss method) (modelzoo.common.pytorch.model_utils.RotaryPositionEmbeddingHelper.RotaryPositionEmbeddingHelper method) (modelzoo.common.pytorch.model_utils.T5ForConditionalGenerationLoss.T5ForConditionalGenerationLoss method) (modelzoo.common.pytorch.run_cstorch_flow.GradScalerParams method) (modelzoo.common.pytorch.utils.BufferedShuffleDataset method) (modelzoo.common.pytorch.utils.IterableDatasetSampler method) (modelzoo.common.pytorch.utils.RunConfigParamsValidator method) (modelzoo.common.pytorch.utils.SampleGenerator method) (modelzoo.fc_mnist.pytorch.model.MNIST method) (modelzoo.fc_mnist.pytorch.model.MNISTModel method) (modelzoo.transformers.data_processing.bert.bertsum_data_processor.BertData method) (modelzoo.transformers.data_processing.bert.bertsum_data_processor.JsonConverter method) (modelzoo.transformers.data_processing.bert.bertsum_data_processor.RougeBasedLabelsFormatter method) (modelzoo.transformers.data_processing.bert.bertsum_data_processor.Tokenizer method) (modelzoo.transformers.data_processing.bert.dynamic_processor.PreprocessInstance method) (modelzoo.transformers.data_processing.bert.mlm_only_processor.MLMOnlyInstance method) (modelzoo.transformers.data_processing.bert.ner_data_processor.InputExample method) (modelzoo.transformers.data_processing.bert.sentence_pair_processor.SentencePairInstance method) (modelzoo.transformers.data_processing.GenericDataProcessor.GenericDataProcessor method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.HDF5Dataset method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.RestartableDataLoader method) (modelzoo.transformers.data_processing.h5_map_dataset.readers.H5Reader method) (modelzoo.transformers.data_processing.h5_map_dataset.readers.Mixture method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.BaseSampler method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.BatchSampler method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.CBSampler method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.Sharder method) (modelzoo.transformers.data_processing.HDF5IterableDataProcessor.HDF5IterableDataProcessor method) (modelzoo.transformers.data_processing.HDF5IterableDataset.HDF5IterableDataset method) (modelzoo.transformers.data_processing.HDF5IterableDataset.RestartableDataLoader method) (modelzoo.transformers.data_processing.huggingface.CSDataCollatorForLanguageModeling.CSDataCollatorForLanguageModeling method) (modelzoo.transformers.data_processing.huggingface.HuggingFaceDataProcessor.HuggingFaceDataProcessor method) (modelzoo.transformers.data_processing.qa.qa_utils.SquadExample method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils.DatasetStats method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils.Reader method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils.VerificationArgs method) (modelzoo.transformers.data_processing.scripts.pubmed.preprocess.Downloader.Downloader method) (modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextFormatting.TextFormatting method) (modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextSharding.NLTKSegmenter method) (modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextSharding.Sharding method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaBooksDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaC4Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaCommonCrawlDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaGithubDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaReplication method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaStackExchangeDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaWikipediaDataset method) (modelzoo.transformers.data_processing.tokenizers.BPETokenizer.BPETokenizer method) (modelzoo.transformers.data_processing.tokenizers.HFTokenizer.HFTokenizer method) (modelzoo.transformers.data_processing.tokenizers.Tokenization.BaseTokenizer method) (modelzoo.transformers.data_processing.tokenizers.Tokenization.FullTokenizer method) (modelzoo.transformers.data_processing.tokenizers.Tokenization.WordPieceTokenizer method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForQuestionAnswering method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForQuestionAnsweringLoss method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSequenceClassification method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSequenceClassificationLoss method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSummarization method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSummarizationLoss method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForTokenClassification method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForTokenClassificationLoss method) (modelzoo.transformers.pytorch.bert.bert_model.BertModel method) (modelzoo.transformers.pytorch.bert.bert_model.BertPooler method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertClassifierHead method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertMLMHead method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertMLMHeadTransform method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertPretrainModel method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.ClassifierDataset method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.DataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.MNLIDataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.MNLIDataset method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.SST2DataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.SST2Dataset method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.model.BertForSequenceClassificationModel method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor.BertSumCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertCSVFormatter method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.model.BertSummarizationModel method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor.BertQADataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.model.BertForQuestionAnsweringModel method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor.BertTokenClassifierDataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.model.BertForTokenClassificationModel method) (modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor.BertCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor.BertCSVDynamicMaskDataProcessor method) (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) (modelzoo.transformers.pytorch.bert.model.BertForPreTrainingModel method) (modelzoo.transformers.pytorch.gpt2.gpt2_model.GPT2LMHeadModel method) (modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor.DummyDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor.DummyDataset method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyIterableDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyIterableDataset method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyTinyIterableDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyTinyIterableDataset method) (modelzoo.transformers.pytorch.gpt2.input.GptHDF5DataProcessor.GptHDF5DataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.GptHDF5MapDataProcessor.GptHDF5MapDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.GptTextDataProcessor.GptTextDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.HuggingFaceDataProcessorEli5.HuggingFaceDataProcessorEli5 method) (modelzoo.transformers.pytorch.gpt2.input.HuggingFaceIterableDataProcessorEli5.HuggingFaceIterableDataProcessorEli5 method) (modelzoo.transformers.pytorch.gpt2.model.Gpt2Model method) (modelzoo.transformers.pytorch.gptj.gptj_model.GPTJModel method) (modelzoo.transformers.pytorch.gptj.model.GptjModel method) (modelzoo.transformers.pytorch.input_utils.ShardedSampler method) (modelzoo.transformers.pytorch.input_utils.SubsetSequentialSampler method) (modelzoo.transformers.pytorch.layers_api_demo.cb_transformer.TransformerModel method) (modelzoo.transformers.pytorch.layers_api_demo.data.AlphabetDataset method) (modelzoo.transformers.pytorch.layers_api_demo.model.TransformerBaseModel method) (modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.PositionalEncoding method) (modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.TransformerModel method) (modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor.T5DynamicDataProcessor method) (modelzoo.transformers.pytorch.t5.model.T5ForConditionalGenerationModel method) (modelzoo.transformers.pytorch.t5.t5_model.T5ForConditionalGeneration method) (modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor.TransformerDynamicDataProcessor method) (modelzoo.transformers.utils.attention_utils.FixedSparseAttentionBuilder method) (modelzoo.transformers.utils.attention_utils.SparseAttentionBuilder method) (modelzoo.transformers.utils.attention_utils.StridedSparseAttentionBuilder method) (modelzoo.vision.pytorch.dit.input.DiffusionBaseProcessor.DiffusionBaseProcessor method) (modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor.CategoricalDataset method) (modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor.DiffusionLatentImageNet1KProcessor method) (modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor.ImageNetLatentDataset method) (modelzoo.vision.pytorch.dit.input.transforms.LabelDropout method) (modelzoo.vision.pytorch.dit.input.transforms.NoiseGenerator method) (modelzoo.vision.pytorch.dit.layers.DiTDecoder.DiTDecoder method) (modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer.DiTDecoderLayer method) (modelzoo.vision.pytorch.dit.layers.GaussianDiffusion.GaussianDiffusion method) (modelzoo.vision.pytorch.dit.layers.RegressionHead.RegressionHead method) (modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer.TimestepEmbeddingLayer method) (modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D.DownEncoderBlock2D method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Downsample2D method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.ResnetBlock2D method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Upsample2D method) (modelzoo.vision.pytorch.dit.layers.vae.UNetMidBlock2D.UNetMidBlock2D method) (modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D.UpDecoderBlock2D method) (modelzoo.vision.pytorch.dit.layers.vae.utils.DiagonalGaussianDistribution method) (modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder.Decoder method) (modelzoo.vision.pytorch.dit.layers.vae.VAEEncoder.Encoder method) (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKLOutput method) (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.DecoderOutput method) (modelzoo.vision.pytorch.dit.model.DiTModel method) (modelzoo.vision.pytorch.dit.modeling_dit.DiT method) (modelzoo.vision.pytorch.dit.pipeline.DiffusionPipeline method) (modelzoo.vision.pytorch.dit.sample_generator.SampleGenerator method) (modelzoo.vision.pytorch.dit.sample_generator_dit.DiTSampleGenerator method) (modelzoo.vision.pytorch.dit.samplers.DDIMSampler.DDIMSampler method) (modelzoo.vision.pytorch.dit.samplers.DDIMSampler.DDIMSamplerOutput method) (modelzoo.vision.pytorch.dit.samplers.DDPMSampler.DDPMSampler method) (modelzoo.vision.pytorch.dit.samplers.DDPMSampler.DDPMSamplerOutput method) (modelzoo.vision.pytorch.input.classification.dataset_factory.Processor method) (modelzoo.vision.pytorch.input.classification.dataset_factory.VisionSubset method) (modelzoo.vision.pytorch.input.classification.mixup.RandomCutmix method) (modelzoo.vision.pytorch.input.classification.mixup.RandomMixup method) (modelzoo.vision.pytorch.input.classification.sampler.RepeatedAugSampler method) (modelzoo.vision.pytorch.input.utils.FastDataLoader method) (modelzoo.vision.pytorch.input.utils.ShardedSampler method) (modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor.Cityscapes method) (modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor.CityscapesDataProcessor method) (modelzoo.vision.pytorch.unet.input.Hdf5BaseDataProcessor.Hdf5BaseDataProcessor method) (modelzoo.vision.pytorch.unet.input.Hdf5BaseIterDataProcessor.Hdf5BaseIterDataProcessor method) (modelzoo.vision.pytorch.unet.input.Hdf5DataProcessor.Hdf5DataProcessor method) (modelzoo.vision.pytorch.unet.input.InriaAerialDataProcessor.InriaAerialDataProcessor method) (modelzoo.vision.pytorch.unet.input.InriaAerialDataProcessor.InriaAerialDataset method) (modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataProcessor method) (modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataset method) (modelzoo.vision.pytorch.unet.input.SkmDataProcessor.SkmDataProcessor method) (modelzoo.vision.pytorch.unet.input.UNetDataProcessor.UNetDataProcessor method) (modelzoo.vision.pytorch.unet.layers.Decoder.Decoder method) (modelzoo.vision.pytorch.unet.layers.Encoder.Encoder method) (modelzoo.vision.pytorch.unet.layers.UNetBlock.UNetBlock method) (modelzoo.vision.pytorch.unet.model.UNetModel method) (modelzoo.vision.pytorch.unet.modeling_unet.UNet method)

__mul__() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures method) (modelzoo.transformers.data_processing.utils.maskedLmInstance method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction.RawResult method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner.InputFeatures method) __new__() (modelzoo.common.pytorch.model_utils.BertPretrainModelLoss.BertPretrainModelLoss static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.ConfigConversionError method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.common.pytorch.model_utils.GPTLMHeadModelLoss.GPTLMHeadModelLoss static method) (modelzoo.common.pytorch.model_utils.T5ForConditionalGenerationLoss.T5ForConditionalGenerationLoss static method) (modelzoo.common.pytorch.utils.BufferedShuffleDataset static method) (modelzoo.common.pytorch.utils.IterableDatasetSampler static method) (modelzoo.fc_mnist.pytorch.model.MNIST static method) (modelzoo.fc_mnist.pytorch.model.MNISTModel static method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.HDF5Dataset static method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.RestartableDataLoader static method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.BaseSampler static method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.BatchSampler static method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.CBSampler static method) (modelzoo.transformers.data_processing.h5_map_dataset.samplers.Sharder static method) (modelzoo.transformers.data_processing.HDF5IterableDataset.HDF5IterableDataset static method) (modelzoo.transformers.data_processing.HDF5IterableDataset.RestartableDataLoader static method) (modelzoo.transformers.data_processing.huggingface.CSDataCollatorForLanguageModeling.CSDataCollatorForLanguageModeling static method) (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures static method) (modelzoo.transformers.data_processing.utils.maskedLmInstance static method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForQuestionAnswering static method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForQuestionAnsweringLoss static method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSequenceClassification static method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSequenceClassificationLoss static method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSummarization static method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForSummarizationLoss static method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForTokenClassification static method) (modelzoo.transformers.pytorch.bert.bert_finetune_models.BertForTokenClassificationLoss static method) (modelzoo.transformers.pytorch.bert.bert_model.BertModel static method) (modelzoo.transformers.pytorch.bert.bert_model.BertPooler static method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertClassifierHead static method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertMLMHead static method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertMLMHeadTransform static method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertPretrainModel static method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.ClassifierDataset static method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.MNLIDataset static method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.SST2Dataset static method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.model.BertForSequenceClassificationModel static method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor.BertSumCSVDataProcessor static method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures static method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.model.BertSummarizationModel static method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor.BertQADataProcessor static method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.model.BertForQuestionAnsweringModel static method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction.RawResult static method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor.BertTokenClassifierDataProcessor static method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner.InputFeatures static method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.model.BertForTokenClassificationModel static method) (modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor.BertCSVDataProcessor static method) (modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor.BertCSVDynamicMaskDataProcessor static method) (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) (modelzoo.transformers.pytorch.bert.model.BertForPreTrainingModel static method) (modelzoo.transformers.pytorch.gpt2.gpt2_model.GPT2LMHeadModel static method) (modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor.DummyDataset static method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyIterableDataset static method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyTinyIterableDataset static method) (modelzoo.transformers.pytorch.gpt2.input.GptTextDataProcessor.GptTextDataProcessor static method) (modelzoo.transformers.pytorch.gpt2.model.Gpt2Model static method) (modelzoo.transformers.pytorch.gptj.gptj_model.GPTJModel static method) (modelzoo.transformers.pytorch.gptj.model.GptjModel static method) (modelzoo.transformers.pytorch.input_utils.ShardedSampler static method) (modelzoo.transformers.pytorch.input_utils.SubsetSequentialSampler static method) (modelzoo.transformers.pytorch.layers_api_demo.cb_transformer.TransformerModel static method) (modelzoo.transformers.pytorch.layers_api_demo.data.AlphabetDataset static method) (modelzoo.transformers.pytorch.layers_api_demo.model.TransformerBaseModel static method) (modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.PositionalEncoding static method) (modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.TransformerModel static method) (modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor.T5DynamicDataProcessor static method) (modelzoo.transformers.pytorch.t5.model.T5ForConditionalGenerationModel static method) (modelzoo.transformers.pytorch.t5.t5_model.T5ForConditionalGeneration static method) (modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor.TransformerDynamicDataProcessor static method) (modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor.CategoricalDataset static method) (modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor.ImageNetLatentDataset static method) (modelzoo.vision.pytorch.dit.input.transforms.LabelDropout static method) (modelzoo.vision.pytorch.dit.input.transforms.NoiseGenerator static method) (modelzoo.vision.pytorch.dit.layers.DiTDecoder.DiTDecoder static method) (modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer.DiTDecoderLayer static method) (modelzoo.vision.pytorch.dit.layers.GaussianDiffusion.GaussianDiffusion static method) (modelzoo.vision.pytorch.dit.layers.RegressionHead.RegressionHead static method) (modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer.TimestepEmbeddingLayer static method) (modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D.DownEncoderBlock2D static method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Downsample2D static method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.ResnetBlock2D static method) (modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D.Upsample2D static method) (modelzoo.vision.pytorch.dit.layers.vae.UNetMidBlock2D.UNetMidBlock2D static method) (modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D.UpDecoderBlock2D static method) (modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder.Decoder static method) (modelzoo.vision.pytorch.dit.layers.vae.VAEEncoder.Encoder static method) (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL static method) (modelzoo.vision.pytorch.dit.model.DiTModel static method) (modelzoo.vision.pytorch.dit.modeling_dit.DiT static method) (modelzoo.vision.pytorch.dit.pipeline.DiffusionPipeline static method) (modelzoo.vision.pytorch.input.classification.dataset_factory.VisionSubset static method) (modelzoo.vision.pytorch.input.classification.mixup.RandomCutmix static method) (modelzoo.vision.pytorch.input.classification.mixup.RandomMixup static method) (modelzoo.vision.pytorch.input.classification.sampler.RepeatedAugSampler static method) (modelzoo.vision.pytorch.input.utils.FastDataLoader static method) (modelzoo.vision.pytorch.input.utils.ShardedSampler static method) (modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor.Cityscapes static method) (modelzoo.vision.pytorch.unet.input.Hdf5BaseDataProcessor.Hdf5BaseDataProcessor static method) (modelzoo.vision.pytorch.unet.input.Hdf5BaseIterDataProcessor.Hdf5BaseIterDataProcessor static method) (modelzoo.vision.pytorch.unet.input.Hdf5DataProcessor.Hdf5DataProcessor static method) (modelzoo.vision.pytorch.unet.input.InriaAerialDataProcessor.InriaAerialDataset static method) (modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor.SeverstalBinaryClassDataset static method) (modelzoo.vision.pytorch.unet.input.SkmDataProcessor.SkmDataProcessor static method) (modelzoo.vision.pytorch.unet.layers.Decoder.Decoder static method) (modelzoo.vision.pytorch.unet.layers.Encoder.Encoder static method) (modelzoo.vision.pytorch.unet.layers.UNetBlock.UNetBlock static method) (modelzoo.vision.pytorch.unet.model.UNetModel static method) (modelzoo.vision.pytorch.unet.modeling_unet.UNet static method) _get_closed_form_lr() (cerebras_pytorch.optim.lr_scheduler.LRScheduler method)

A

AccuracyMetric (class in cerebras_pytorch.metrics) Adadelta (class in cerebras_pytorch.optim) Adafactor (class in cerebras_pytorch.optim) Adagrad (class in cerebras_pytorch.optim) AdaLayerNorm (class in modelzoo.common.pytorch.layers) Adam (class in cerebras_pytorch.optim) Adamax (class in cerebras_pytorch.optim) AdamBase (in module cerebras_pytorch.optim) AdamW (class in cerebras_pytorch.optim) add_common_args() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) add_common_parser_args() (in module modelzoo.transformers.pytorch.bert.input.scripts.parser_utils) add_csx_arguments() (in module modelzoo.common.run_utils.cli_parser) add_general_arguments() (in module modelzoo.common.run_utils.cli_parser) add_gpu_arguments() (in module modelzoo.common.run_utils.cli_parser) add_mlm_nsp_specific_args() (in module modelzoo.transformers.pytorch.bert.input.scripts.parser_utils) add_mlm_only_specific_args() (in module modelzoo.transformers.pytorch.bert.input.scripts.parser_utils) add_tensor() (cerebras_pytorch.utils.tensorboard.SummaryWriter method) add_token() (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method)

adjust_brightness_transform() (in module modelzoo.vision.pytorch.unet.input.preprocessing_utils) aggregate_state_dict() (cerebras_pytorch.utils.data.RestartableDataLoader method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.RestartableDataLoader method) AlibiPositionEmbeddingLayer (class in modelzoo.common.pytorch.layers) AlphabetDataset (class in modelzoo.transformers.pytorch.layers_api_demo.data) already_shuffled() (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaBooksDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaC4Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaCommonCrawlDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaGithubDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaReplication method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaStackExchangeDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaWikipediaDataset method) annotate_sparsity() (cerebras_pytorch.sparse.base.BaseSparsityOptimizer method) append() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) apply_sparsity() (cerebras_pytorch.sparse.base.BaseSparsityOptimizer method) apply_transform_to_files() (in module modelzoo.vision.pytorch.unet.input.scripts.resize_inria_files) arg_filter() (in module modelzoo.common.pytorch.run_utils) ASGD (class in cerebras_pytorch.optim) assemble_disallowlist() (in module modelzoo.common.run_utils.cli_parser) AutoencoderKL (class in modelzoo.vision.pytorch.dit.layers.vae.VAEModel) AutoencoderKLOutput (class in modelzoo.vision.pytorch.dit.layers.vae.VAEModel)

B

backend() (in module cerebras_pytorch) BaseCheckpointConverter (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) BaseCheckpointConverter_CS_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) BaseCheckpointConverter_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) BaseConfigConverter (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) BaseConfigConverter_CS_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) BaseConfigConverter_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) BaseDictionaryConverter (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) BaseHyperParameter (class in cerebras_pytorch.sparse.utils) BaseSampler (class in modelzoo.transformers.data_processing.h5_map_dataset.samplers) BaseSchedule (class in cerebras_pytorch.sparse.dynamic) BaseSparsityOptimizer (class in cerebras_pytorch.sparse.base) BaseTokenizer (class in modelzoo.transformers.data_processing.tokenizers.Tokenization) batch_metrics (cerebras_pytorch.utils.benchmark.utils.dataloader.EpochMetrics attribute) batch_specs (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics attribute) BatchChannelNorm2D (class in modelzoo.common.pytorch.layers) BatchMetrics (class in cerebras_pytorch.utils.benchmark.utils.dataloader) BatchSampler (class in modelzoo.transformers.data_processing.h5_map_dataset.samplers) benchmark_dataloader() (in module cerebras_pytorch.utils.benchmark) BertClassifierHead (class in modelzoo.transformers.pytorch.bert.bert_pretrain_models) BertCSVDataProcessor (class in modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor) BertCSVDynamicMaskDataProcessor (class in modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor) BertCSVFormatter (class in modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils) BertData (class in modelzoo.transformers.data_processing.bert.bertsum_data_processor) BertForPreTrainingModel (class in modelzoo.transformers.pytorch.bert.model) BertForQuestionAnswering (class in modelzoo.transformers.pytorch.bert.bert_finetune_models) BertForQuestionAnsweringLoss (class in modelzoo.transformers.pytorch.bert.bert_finetune_models) BertForQuestionAnsweringModel (class in modelzoo.transformers.pytorch.bert.fine_tuning.qa.model)

BertForSequenceClassification (class in modelzoo.transformers.pytorch.bert.bert_finetune_models) BertForSequenceClassificationLoss (class in modelzoo.transformers.pytorch.bert.bert_finetune_models) BertForSequenceClassificationModel (class in modelzoo.transformers.pytorch.bert.fine_tuning.classifier.model) BertForSummarization (class in modelzoo.transformers.pytorch.bert.bert_finetune_models) BertForSummarizationLoss (class in modelzoo.transformers.pytorch.bert.bert_finetune_models) BertForTokenClassification (class in modelzoo.transformers.pytorch.bert.bert_finetune_models) BertForTokenClassificationLoss (class in modelzoo.transformers.pytorch.bert.bert_finetune_models) BertForTokenClassificationModel (class in modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.model) BertInputFeatures (class in modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils) BertMLMHead (class in modelzoo.transformers.pytorch.bert.bert_pretrain_models) BertMLMHeadTransform (class in modelzoo.transformers.pytorch.bert.bert_pretrain_models) BertModel (class in modelzoo.transformers.pytorch.bert.bert_model) BertPooler (class in modelzoo.transformers.pytorch.bert.bert_model) BertPretrainModel (class in modelzoo.transformers.pytorch.bert.bert_pretrain_models) BertPretrainModelLoss (class in modelzoo.common.pytorch.model_utils.BertPretrainModelLoss) BertQADataProcessor (class in modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor) BertSumCSVDataProcessor (class in modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor) BertSummarizationModel (class in modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.model) BertTokenClassifierDataProcessor (class in modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor) BiaslessLayerNorm (class in modelzoo.common.pytorch.layers) BlockType (class in modelzoo.vision.pytorch.dit.utils) BPETokenizer (class in modelzoo.transformers.data_processing.tokenizers.BPETokenizer) bucketed_batch() (in module modelzoo.common.pytorch.input_utils) BufferedShuffleDataset (class in modelzoo.common.pytorch.utils) build_broadcastable_attention_mask() (in module modelzoo.transformers.pytorch.transformer_utils) build_inputs() (modelzoo.vision.pytorch.dit.pipeline.DiffusionPipeline method) build_vocab() (in module modelzoo.transformers.pytorch.bert.input.utils) bytes_to_unicode() (in module modelzoo.transformers.data_processing.tokenizers.BPETokenizer)

C

CategoricalDataset (class in modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor) CBSampler (class in modelzoo.transformers.data_processing.h5_map_dataset.samplers) cerebras_pytorch.sparse.base module cerebras_pytorch.sparse.dynamic module cerebras_pytorch.sparse.init module cerebras_pytorch.sparse.utils module cerebras_pytorch.utils.CSConfig (built-in class) cerebras_pytorch.utils.data.DataLoaderCheckpoint (built-in class) ChainedScheduler (class in cerebras_pytorch.optim.lr_scheduler) check_and_create_output_dirs() (in module modelzoo.common.input.utils) check_default_args() (in module modelzoo.common.pytorch.model_utils.convert_config_to_mup) check_output() (in module modelzoo.transformers.data_processing.bert.bertsum_data_processor) check_sharding_sanity() (in module modelzoo.transformers.pytorch.input_utils) check_tokenizer_scheme() (in module modelzoo.transformers.data_processing.qa.qa_utils) check_unused_model_params() (in module modelzoo.transformers.pytorch.bert.utils) checkerboard() (in module cerebras_pytorch.sparse.init) checkpoint_closure() (in module cerebras_pytorch) CheckpointConverterCLI (class in modelzoo.common.pytorch.model_utils.convert_checkpoint) Cityscapes (class in modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor) CityscapesDataProcessor (class in modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor) ClassifierDataset (class in modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor) clean() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.filter) clear() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) clip_gradients_and_return_isfinite() (cerebras_pytorch.amp.GradScaler method) cls_ids (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures attribute) cluster_config() (in module modelzoo.transformers.pytorch.input_utils) collect_stats() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) compile() (in module cerebras_pytorch) compute() (cerebras_pytorch.metrics.AccuracyMetric method) (cerebras_pytorch.metrics.DiceCoefficientMetric method) (cerebras_pytorch.metrics.FBetaScoreMetric method) (cerebras_pytorch.metrics.MeanIOUMetric method) (cerebras_pytorch.metrics.Metric method) (cerebras_pytorch.metrics.PerplexityMetric method) compute_all_metrics() (in module cerebras_pytorch.metrics) compute_grad_norm() (in module modelzoo.common.pytorch.run_cstorch_flow) compute_params_norm() (in module modelzoo.common.pytorch.run_cstorch_flow) concatenate_documents() (in module modelzoo.transformers.pytorch.t5.input.utils) ConfigConversionError ConfigConverter_Bert_CS16_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) ConfigConverter_Bert_CS16_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) ConfigConverter_Bert_CS17_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) ConfigConverter_Bert_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) ConfigConverter_Bert_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) ConfigConverter_BertForQuestionAnswering_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) ConfigConverter_BertForQuestionAnswering_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) ConfigConverter_BertForSequenceClassification_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) ConfigConverter_BertForSequenceClassification_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) ConfigConverter_BertForTokenClassification_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) ConfigConverter_BertForTokenClassification_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) ConfigConverter_BloomLMHeadModel_CS19_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) ConfigConverter_BloomModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) ConfigConverter_BloomModel_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) ConfigConverter_BloomModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) ConfigConverter_BTLMModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs) ConfigConverter_Codegen_Model_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) ConfigConverter_Codegen_Model_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) ConfigConverter_Codegen_Model_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) ConfigConverter_Codegen_Model_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) ConfigConverter_Falcon_40B_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b) ConfigConverter_Falcon_7B_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b) ConfigConverter_Falcon_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon) ConfigConverter_GPT2Model_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) ConfigConverter_GPT2Model_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) ConfigConverter_GPT2Model_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) ConfigConverter_GPT2Model_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) ConfigConverter_GPT_Neox_Headless_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) ConfigConverter_GPT_Neox_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) ConfigConverter_GPT_Neox_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) ConfigConverter_GPT_Neox_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) ConfigConverter_GPTJModel_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) ConfigConverter_GPTJModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) ConfigConverter_GPTJModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) ConfigConverter_GPTJModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) ConfigConverter_LLaMa_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) ConfigConverter_LLaMa_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) ConfigConverter_LlamaModel_CS19_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) ConfigConverter_MPT_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt) ConfigConverter_OPTModel_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) ConfigConverter_OPTModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) ConfigConverter_OPTModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) ConfigConverter_OPTModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) ConfigConverter_Roberta_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta) ConfigConverter_SantacoderModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder) ConfigConverter_StarcoderModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder) ConfigConverter_T5_CS16_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) ConfigConverter_T5_CS16_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) ConfigConverter_T5_CS17_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) ConfigConverter_T5_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) ConfigConverter_T5_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) ConfigConverter_T5_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) ConfigConverter_T5_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) ConfigConverter_ViT_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.vit) ConfigConverter_ViTMAE_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae) configure_lr_scheduler() (in module cerebras_pytorch.optim) configure_optimizer() (in module cerebras_pytorch.optim) configure_sparsity_optimizer() (in module cerebras_pytorch.sparse) configure_sparsity_wrapper() (in module cerebras_pytorch.sparse) ConstantHyperParameter (class in cerebras_pytorch.sparse.utils) ConstantLR (class in cerebras_pytorch.optim.lr_scheduler) construct_denoising_objective() (in module modelzoo.transformers.pytorch.t5.input.utils) construct_graph() (in module modelzoo.transformers.data_processing.slimpajama.dedup.generate_connected_components) ConversionRule (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) convert_checkpoint() (in module modelzoo.common.pytorch.model_utils.convert_checkpoint) convert_checkpoint_from_file() (in module modelzoo.common.pytorch.model_utils.convert_checkpoint) convert_config() (in module modelzoo.common.pytorch.model_utils.convert_checkpoint) convert_config_from_file() (in module modelzoo.common.pytorch.model_utils.convert_checkpoint) convert_dataset_to_HDF5() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.convert_dataset_to_HDF5) convert_examples_to_features() (in module modelzoo.transformers.data_processing.qa.qa_utils) convert_examples_to_features_and_write() (in module modelzoo.transformers.data_processing.qa.qa_utils) (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner) convert_file_size_to_int() (in module modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints) convert_helper() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.ConfigConverter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.ConfigConverter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.ConfigConverter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.ConfigConverter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_Headless_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LLaMa_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LLaMa_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LlamaModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.ConfigConverter_MPT_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.ConfigConverter_Roberta_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.ConfigConverter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.ConfigConverter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.ConfigConverter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.ConfigConverter_ViTMAE_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 method) convert_ids_to_tokens() (modelzoo.transformers.data_processing.tokenizers.Tokenization.FullTokenizer method) convert_key() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseDictionaryConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.ConfigConverter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.ConfigConverter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.ConfigConverter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.ConfigConverter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_Headless_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LLaMa_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LLaMa_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LlamaModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.ConfigConverter_MPT_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.ConfigConverter_Roberta_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.ConfigConverter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.ConfigConverter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.ConfigConverter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.ConfigConverter_ViTMAE_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 method)

convert_single_example() (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner) convert_str_to_int_list() (in module modelzoo.transformers.data_processing.utils) convert_to_bert_format_files() (in module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.write_csv_data) convert_to_json_files() (in module modelzoo.transformers.data_processing.bert.bertsum_data_processor) convert_to_unicode() (in module modelzoo.transformers.data_processing.utils) (in module modelzoo.transformers.pytorch.bert.input.utils) convert_tokens_to_ids() (modelzoo.transformers.data_processing.tokenizers.Tokenization.FullTokenizer method) convert_use_biasless_layer_norm_helper() (in module modelzoo.common.pytorch.model_utils.checkpoint_converters.helper) convert_use_rms_layer_norm_helper() (in module modelzoo.common.pytorch.model_utils.checkpoint_converters.helper) Converter_Bert_CS17_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertFinetuneModel_CS16_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) Converter_BertFinetuneModel_CS16_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) Converter_BertForQuestionAnswering_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) Converter_BertForQuestionAnswering_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) Converter_BertForSequenceClassification_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) Converter_BertForSequenceClassification_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) Converter_BertForTokenClassification_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) Converter_BertForTokenClassification_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune) Converter_BertLayerNorm_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertModel_CS16_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertModel_CS16_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertPretrainModel_CS16_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertPretrainModel_CS16_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertPretrainModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_BertPretrainModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bert) Converter_Bloom_Attention_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) Converter_BloomLMHeadModel_CS19_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) Converter_BloomLMHeadModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) Converter_BloomLMHeadModel_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) Converter_BloomLMHeadModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) Converter_BloomModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) Converter_BloomModel_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) Converter_BloomModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs) Converter_BTLMLMHeadModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs) Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs) Converter_BTLMModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs) Converter_BTLMModel_WithoutModelPrefix_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs) Converter_Codegen_Attention_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) Converter_Codegen_Headless_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) Converter_Codegen_Headless_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) Converter_Codegen_Headless_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) Converter_Codegen_LMHeadModel_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) Converter_Codegen_LMHeadModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) Converter_Codegen_LMHeadModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) Converter_Codegen_LMHeadModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs) Converter_Falcon_40B_Attention_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b) Converter_Falcon_40B_Headless_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b) Converter_Falcon_40B_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b) Converter_Falcon_7B_Attention_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b) Converter_Falcon_7B_Headless_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b) Converter_Falcon_7B_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b) Converter_Falcon_Headless_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon) Converter_Falcon_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon) Converter_GPT2_Attention_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) Converter_GPT2LMHeadModel_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) Converter_GPT2LMHeadModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) Converter_GPT2LMHeadModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) Converter_GPT2LMHeadModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) Converter_GPT2Model_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) Converter_GPT2Model_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) Converter_GPT2Model_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) Converter_GPT_Neox_Attention_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) Converter_GPT_Neox_Headless_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) Converter_GPT_Neox_Headless_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) Converter_GPT_Neox_Headless_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) Converter_GPT_Neox_LMHeadModel_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) Converter_GPT_Neox_LMHeadModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) Converter_GPT_Neox_LMHeadModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) Converter_GPT_Neox_LMHeadModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs) Converter_GPTJ_Attention_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) Converter_GPTJ_Headless_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) Converter_GPTJ_Headless_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) Converter_GPTJ_Headless_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) Converter_GPTJ_LMHeadModel_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) Converter_GPTJ_LMHeadModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) Converter_GPTJ_LMHeadModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) Converter_GPTJ_LMHeadModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs) Converter_LlamaAttention_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) Converter_LlamaForCausalLM_CS19_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) Converter_LlamaForCausalLM_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) Converter_LlamaForCausalLM_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) Converter_LlamaForCausalLM_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) Converter_LlamaModel_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) Converter_LlamaModel_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) Converter_LlamaModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.llama) Converter_MPTAttention_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt) Converter_MPTForCausalLM_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt) Converter_MPTForCausalLM_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt) Converter_MPTModel_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt) Converter_MPTModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt) Converter_OPT_Attention_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) Converter_OPT_Headless_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) Converter_OPT_Headless_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) Converter_OPT_Headless_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) Converter_OPT_LMHeadModel_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) Converter_OPT_LMHeadModel_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) Converter_OPT_LMHeadModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) Converter_OPT_LMHeadModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs) Converter_RobertaPretrainModel_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta) Converter_RobertaPretrainModel_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta) Converter_Santacoder_Attention_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder) Converter_SantacoderLMHeadModel_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder) Converter_SantacoderLMHeadModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder) Converter_SantacoderModel_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder) Converter_SantacoderModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder) Converter_Starcoder_Attention_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder) Converter_StarcoderForCausalLM_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder) Converter_StarcoderForCausalLM_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder) Converter_StarcoderModel_HF_CS (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder) Converter_StarcoderModel_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder) Converter_T5_CS16_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) Converter_T5_CS16_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) Converter_T5_CS17_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) Converter_T5_CS18_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) Converter_T5_HF_CS17 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) Converter_T5_HF_CS18 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) Converter_T5_HF_CS20 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.t5) Converter_ViT_Core_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.vit) Converter_ViT_Headless_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.vit) Converter_ViT_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.vit) Converter_ViTMAE_Core_HF_CS19 (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae) copy() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) CosineAnnealingLR (class in cerebras_pytorch.optim.lr_scheduler) CosineAnnealingWarmRestarts (class in cerebras_pytorch.optim.lr_scheduler) CosineDecayLR (class in cerebras_pytorch.optim.lr_scheduler) CosineHyperParameter (class in cerebras_pytorch.sparse.utils) count() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures method) (modelzoo.transformers.data_processing.utils.maskedLmInstance method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction.RawResult method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner.InputFeatures method) count_lines() (in module modelzoo.common.model_utils.count_lines) count_total_documents() (in module modelzoo.transformers.data_processing.utils) create_2D_autoregressive_mask() (in module modelzoo.transformers.pytorch.transformer_utils) create_2D_full_mask() (in module modelzoo.transformers.pytorch.transformer_utils) create_arg_parser() (in module modelzoo.transformers.pytorch.bert.input.scripts.parser_utils) create_bertsum_feature() (in module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor) create_csv() (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only_static_masking) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_static_masking) create_dataloader() (modelzoo.transformers.data_processing.GenericDataProcessor.GenericDataProcessor method) (modelzoo.transformers.data_processing.HDF5IterableDataProcessor.HDF5IterableDataProcessor method) (modelzoo.transformers.data_processing.huggingface.HuggingFaceDataProcessor.HuggingFaceDataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor.BertSumCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor.BertQADataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor.BertTokenClassifierDataProcessor method) (modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor.BertCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor.BertCSVDynamicMaskDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor.DummyDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyIterableDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor.DummyTinyIterableDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.GptHDF5DataProcessor.GptHDF5DataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.GptTextDataProcessor.GptTextDataProcessor method) (modelzoo.transformers.pytorch.gpt2.input.HuggingFaceDataProcessorEli5.HuggingFaceDataProcessorEli5 method) (modelzoo.transformers.pytorch.gpt2.input.HuggingFaceIterableDataProcessorEli5.HuggingFaceIterableDataProcessorEli5 method) (modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor.T5DynamicDataProcessor method) (modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor.TransformerDynamicDataProcessor method) (modelzoo.vision.pytorch.dit.input.DiffusionBaseProcessor.DiffusionBaseProcessor method) (modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor.DiffusionLatentImageNet1KProcessor method) (modelzoo.vision.pytorch.unet.input.Hdf5BaseDataProcessor.Hdf5BaseDataProcessor method) (modelzoo.vision.pytorch.unet.input.Hdf5BaseIterDataProcessor.Hdf5BaseIterDataProcessor method) (modelzoo.vision.pytorch.unet.input.Hdf5DataProcessor.Hdf5DataProcessor method) (modelzoo.vision.pytorch.unet.input.SkmDataProcessor.SkmDataProcessor method) create_dataset() (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method) create_diffusion_model() (modelzoo.vision.pytorch.dit.sample_generator.SampleGenerator method) (modelzoo.vision.pytorch.dit.sample_generator_dit.DiTSampleGenerator method) create_features_auto_lm() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) create_features_summarization() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) create_fix_pos_embedding() (modelzoo.common.pytorch.layers.EmbeddingLayer method) create_fixed_sparse_attention_mask() (in module modelzoo.transformers.pytorch.gpt2.sparse_mask) create_h5() (in module modelzoo.transformers.pytorch.bert.input.scripts.create_hdf5_files) (in module modelzoo.vision.pytorch.unet.input.scripts.create_hdf5_files) create_h5_mp() (in module modelzoo.transformers.pytorch.bert.input.scripts.create_hdf5_files) (in module modelzoo.vision.pytorch.unet.input.scripts.create_hdf5_files) create_initializer() (in module modelzoo.common.pytorch.model_utils.create_initializer) create_masked_lm_features() (in module modelzoo.transformers.data_processing.bert.mlm_only_processor) create_masked_lm_predictions() (in module modelzoo.transformers.data_processing.utils) (in module modelzoo.transformers.pytorch.bert.input.utils) create_ner_features() (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor) create_npz_from_sample_folder() (in module modelzoo.vision.pytorch.dit.sample_generator) create_parser() (in module modelzoo.transformers.data_processing.bert.bertsum_data_processor) (in module modelzoo.transformers.data_processing.bert.ner_data_processor) create_pipeline() (modelzoo.vision.pytorch.dit.sample_generator.SampleGenerator method) (modelzoo.vision.pytorch.dit.sample_generator_dit.DiTSampleGenerator method) create_preprocessing_params_with_defaults() (in module modelzoo.vision.pytorch.input.classification.utils) create_qa_features() (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor) create_sampler() (modelzoo.vision.pytorch.dit.sample_generator.SampleGenerator method) (modelzoo.vision.pytorch.dit.sample_generator_dit.DiTSampleGenerator method) create_timestep_embedding() (modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer.TimestepEmbeddingLayer static method) create_transform() (in module modelzoo.vision.pytorch.input.transforms) create_transformer_input_features() (in module modelzoo.transformers.pytorch.t5.input.utils) create_vae_model() (modelzoo.vision.pytorch.dit.sample_generator_dit.DiTSampleGenerator method) create_worker_cache() (in module modelzoo.vision.pytorch.input.utils) CSDataCollatorForLanguageModeling (class in modelzoo.transformers.data_processing.huggingface.CSDataCollatorForLanguageModeling) CSoftPackage (class in modelzoo) CurationCorpusPreprocessor (class in modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor) current_backend() (in module cerebras_pytorch) current_torch_device() (in module cerebras_pytorch) cycle_documents() (in module modelzoo.transformers.data_processing.slimpajama.utils.utils) CyclicLR (class in cerebras_pytorch.optim.lr_scheduler) CyclingHyperParameter (class in cerebras_pytorch.sparse.utils)

D

data_generator() (in module modelzoo.transformers.data_processing.bert.dynamic_processor) (in module modelzoo.transformers.data_processing.bert.mlm_only_processor) (in module modelzoo.transformers.data_processing.bert.sentence_pair_processor) DataExecutor (class in cerebras_pytorch.utils.data) DataLoader (class in cerebras_pytorch.utils.data) dataloader_build_time (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics attribute) DataProcessor (class in modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor) Dataset (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) DatasetStats (class in modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) DDIMSampler (class in modelzoo.vision.pytorch.dit.samplers.DDIMSampler) DDIMSamplerOutput (class in modelzoo.vision.pytorch.dit.samplers.DDIMSampler) DDPMSampler (class in modelzoo.vision.pytorch.dit.samplers.DDPMSampler) DDPMSamplerOutput (class in modelzoo.vision.pytorch.dit.samplers.DDPMSampler) deaggregate_state_dict() (cerebras_pytorch.utils.data.RestartableDataLoader method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.RestartableDataLoader method) debug_or_download_individual_file() (in module modelzoo.transformers.data_processing.scripts.pile.download) Decoder (class in modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder) (class in modelzoo.vision.pytorch.unet.layers.Decoder) DecoderOutput (class in modelzoo.vision.pytorch.dit.layers.vae.VAEModel) deduplicate_train_holdout_sets() (in module modelzoo.transformers.data_processing.slimpajama.dedup.dedup_train) devices() (modelzoo.common.run_utils.utils.DeviceType class method) DeviceType (class in modelzoo.common.run_utils.utils) DiagonalGaussianDistribution (class in modelzoo.vision.pytorch.dit.layers.vae.utils) DiceCoefficientMetric (class in cerebras_pytorch.metrics) diff_checkpoints() (in module modelzoo.common.pytorch.model_utils.convert_checkpoint) diff_checkpoints_from_file() (in module modelzoo.common.pytorch.model_utils.convert_checkpoint) DiffusionBaseProcessor (class in modelzoo.vision.pytorch.dit.input.DiffusionBaseProcessor) DiffusionLatentImageNet1KProcessor (class in modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor) DiffusionPipeline (class in modelzoo.vision.pytorch.dit.pipeline) dir_path() (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaBooksDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaC4Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaCommonCrawlDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaGithubDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaReplication method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaStackExchangeDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaWikipediaDataset method)

disable_collection() (modelzoo.common.pytorch.dump_context.DumpContext method) disable_dataloader_checkpointing() (cerebras_pytorch.utils.data.DataLoader method) disable_slicing() (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) disable_tiling() (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) discard_params() (in module modelzoo.common.run_utils.cli_parser) display_images() (in module modelzoo.vision.pytorch.dit.display_images) DiT (class in modelzoo.vision.pytorch.dit.modeling_dit) DiTDecoder (class in modelzoo.vision.pytorch.dit.layers.DiTDecoder) DiTDecoderLayer (class in modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer) DiTModel (class in modelzoo.vision.pytorch.dit.model) DiTSampleGenerator (class in modelzoo.vision.pytorch.dit.sample_generator_dit) doc_span_index (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) documents() (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaBooksDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaC4Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaCommonCrawlDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaGithubDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaReplication method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaStackExchangeDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaWikipediaDataset method) DownEncoderBlock2D (class in modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D) download_pile() (in module modelzoo.transformers.data_processing.scripts.pile.download) download_tokenizer_files() (in module modelzoo.transformers.data_processing.scripts.pile.download) Downloader (class in modelzoo.transformers.data_processing.scripts.pubmed.preprocess.Downloader) Downsample2D (class in modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D) dtype_byte_size() (in module modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints) DummyDataProcessor (class in modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor) DummyDataset (class in modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor) DummyIterableDataProcessor (class in modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor) DummyIterableDataset (class in modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor) DummyTinyIterableDataProcessor (class in modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor) DummyTinyIterableDataset (class in modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor) dump_args() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) dump_result() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) DumpContext (class in modelzoo.common.pytorch.dump_context) DynamicSparsityOptimizer (class in cerebras_pytorch.sparse.dynamic)

E

element_length_fn() (modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor.T5DynamicDataProcessor method) (modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor.TransformerDynamicDataProcessor method) EmbeddingLayer (class in modelzoo.common.pytorch.layers) enable_collection() (modelzoo.common.pytorch.dump_context.DumpContext method) enable_slicing() (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) enable_tiling() (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) encode_sequence() (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.ClassifierDataset method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.MNLIDataset method) (modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor.SST2Dataset method) Encoder (class in modelzoo.vision.pytorch.dit.layers.vae.VAEEncoder) (class in modelzoo.vision.pytorch.unet.layers.Encoder) end_logits (modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction.RawResult attribute) end_position (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) end_time_ns (cerebras_pytorch.utils.benchmark.utils.dataloader.EpochMetrics attribute) (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics attribute) epoch_metrics (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics attribute) epoch_step (cerebras_pytorch.utils.benchmark.utils.dataloader.BatchMetrics attribute) EpochMetrics (class in cerebras_pytorch.utils.benchmark.utils.dataloader) EquivalentSubkey (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) eval_input_dataloader() (in module modelzoo.transformers.pytorch.bert.data) (in module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.data) (in module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.data) (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.data) (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.data) (in module modelzoo.transformers.pytorch.gpt2.data) (in module modelzoo.transformers.pytorch.layers_api_demo.data) (in module modelzoo.transformers.pytorch.t5.data) (in module modelzoo.transformers.pytorch.transformer.data) (in module modelzoo.vision.pytorch.dit.data) (in module modelzoo.vision.pytorch.unet.data) example_index (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) ExpHyperParameter (class in cerebras_pytorch.sparse.utils) ExponentialLR (class in cerebras_pytorch.optim.lr_scheduler) extend() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) extract() (in module modelzoo.vision.pytorch.dit.layers.GaussianDiffusion) extract_model_dict() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 method)

F

FastDataLoader (class in modelzoo.vision.pytorch.input.utils) FBetaScoreMetric (class in cerebras_pytorch.metrics) FeedForwardNetwork (class in modelzoo.common.pytorch.layers) file_read_generator() (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method) filter_dataset() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.filter) find_connected_components() (in module modelzoo.transformers.data_processing.slimpajama.dedup.generate_connected_components) FirstScoreTiebreaker (class in cerebras_pytorch.sparse.utils) FixedSparseAttentionBuilder (class in modelzoo.transformers.utils.attention_utils) flat_map() (in module modelzoo.transformers.pytorch.t5.input.utils) flush() (modelzoo.common.pytorch.dump_context.DumpContext method) FormatVersions (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter) forward() (cerebras_pytorch.metrics.Metric method) (modelzoo.common.pytorch.layers.AlibiPositionEmbeddingLayer method) (modelzoo.common.pytorch.layers.EmbeddingLayer method) (modelzoo.common.pytorch.layers.GPTJDecoderLayer method) (modelzoo.common.pytorch.layers.MultiheadAttention method) (modelzoo.common.pytorch.layers.RelativePositionEmbeddingLayer method) (modelzoo.common.pytorch.layers.Transformer method) (modelzoo.common.pytorch.layers.TransformerDecoder method) (modelzoo.common.pytorch.layers.TransformerDecoderLayer method) (modelzoo.common.pytorch.layers.TransformerEncoder method) (modelzoo.common.pytorch.layers.TransformerEncoderLayer method) (modelzoo.common.pytorch.layers.ViTEmbeddingLayer method) (modelzoo.common.pytorch.model_utils.T5ForConditionalGenerationLoss.T5ForConditionalGenerationLoss method) (modelzoo.transformers.pytorch.bert.bert_model.BertModel method) (modelzoo.transformers.pytorch.bert.bert_pretrain_models.BertPretrainModel method) (modelzoo.transformers.pytorch.layers_api_demo.cb_transformer.TransformerModel method) (modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.PositionalEncoding method) (modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer.TransformerModel method) (modelzoo.transformers.pytorch.t5.t5_model.T5ForConditionalGeneration method) (modelzoo.vision.pytorch.dit.input.transforms.NoiseGenerator method) (modelzoo.vision.pytorch.dit.layers.DiTDecoder.DiTDecoder method) (modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer.DiTDecoderLayer method) (modelzoo.vision.pytorch.dit.layers.GaussianDiffusion.GaussianDiffusion method) (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) (modelzoo.vision.pytorch.dit.pipeline.DiffusionPipeline method) (modelzoo.vision.pytorch.input.classification.mixup.RandomCutmix method) (modelzoo.vision.pytorch.input.classification.mixup.RandomMixup method)

forward_dit_with_cfg() (modelzoo.vision.pytorch.dit.modeling_dit.DiT method) FreqSchedule (class in cerebras_pytorch.sparse.dynamic) from_dict() (modelzoo.common.pytorch.run_cstorch_flow.GradScalerParams class method) from_numpy() (in module cerebras_pytorch) from_zeros() (in module cerebras_pytorch.sparse.init) fromkeys() (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) full() (in module cerebras_pytorch) full_like() (in module cerebras_pytorch) FullTokenizer (class in modelzoo.transformers.data_processing.tokenizers.Tokenization)

G

GaussianDiffusion (class in modelzoo.vision.pytorch.dit.layers.GaussianDiffusion) geglu() (in module modelzoo.common.pytorch.model_utils.activations) gelu_fast() (in module modelzoo.common.pytorch.model_utils.activations) gelu_new() (in module modelzoo.common.pytorch.model_utils.activations) generate_connected_components_mp() (in module modelzoo.transformers.data_processing.slimpajama.dedup.generate_connected_components) generate_duplicates() (in module modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicates_dict) generate_hashes() (in module modelzoo.transformers.data_processing.slimpajama.dedup.to_hash) generate_mup_config() (in module modelzoo.common.pytorch.model_utils.convert_config_to_mup) generate_pairs() (in module modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicate_pairs) generate_samples() (in module modelzoo.transformers.data_processing.slimpajama.split_dataset) generate_square_subsequent_mask() (in module modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer) GenericDataProcessor (class in modelzoo.transformers.data_processing.GenericDataProcessor) get() (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) get_activation() (in module modelzoo.common.pytorch.model_utils.activations) get_all_args() (in module modelzoo.common.run_utils.cli_parser) get_beta_schedule() (in module modelzoo.vision.pytorch.dit.layers.schedulers) get_checkpoints() (in module modelzoo.common.pytorch.utils) get_cls() (cerebras_pytorch.sparse.dynamic.BaseSchedule static method) (cerebras_pytorch.sparse.utils.BaseHyperParameter static method) get_data_for_task() (in module modelzoo.transformers.pytorch.input_utils) get_default_inis() (in module modelzoo.vision.pytorch.utils.run_utils) get_documents() (in module modelzoo.transformers.data_processing.slimpajama.dedup.to_hash) get_eval_dataloader() (in module modelzoo.fc_mnist.pytorch.data) get_extended_attention_mask() (in module modelzoo.transformers.pytorch.transformer_utils) get_features() (in module modelzoo.transformers.data_processing.slimpajama.dedup.to_hash) get_files() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) get_files_in_metadata() (in module modelzoo.transformers.data_processing.utils) get_final_text() (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction) get_hashes() (in module modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicate_pairs) get_label_id_map() (in module modelzoo.transformers.data_processing.utils) get_latest_checkpoint() (in module modelzoo.common.pytorch.run_cstorch_flow) get_lr() (cerebras_pytorch.optim.lr_scheduler.LRScheduler method) get_meta_data() (in module modelzoo.transformers.pytorch.bert.input.utils) (modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor.T5DynamicDataProcessor method) (modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor.TransformerDynamicDataProcessor method) get_min_max_end() (cerebras_pytorch.sparse.utils.BaseHyperParameter method) (cerebras_pytorch.sparse.utils.CosineHyperParameter method) (cerebras_pytorch.sparse.utils.CyclingHyperParameter method) (cerebras_pytorch.sparse.utils.LambdaHyperParameter method) get_model_checkpoint() (in module modelzoo.common.pytorch.run_cstorch_flow) get_model_converter() (in module modelzoo.common.pytorch.model_utils.convert_checkpoint) get_named_beta_schedule() (in module modelzoo.vision.pytorch.dit.layers.schedulers) get_norm() (in module modelzoo.common.pytorch.model_utils.norms) get_output_type_shapes() (in module modelzoo.transformers.data_processing.utils) get_pairs() (in module modelzoo.transformers.data_processing.tokenizers.BPETokenizer) get_params() (in module modelzoo.common.run_utils.cli_parser) (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) get_params_from_args() (in module modelzoo.common.run_utils.cli_parser) (in module modelzoo.common.run_utils.cli_pytorch) get_parser() (in module modelzoo.common.run_utils.cli_parser) (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils)

get_parser_args() (in module modelzoo.transformers.pytorch.bert.input.scripts.parser_utils) (in module modelzoo.vision.pytorch.dit.display_images) (in module modelzoo.vision.pytorch.dit.sample_generator_dit) (in module modelzoo.vision.pytorch.unet.input.scripts.create_hdf5_files) (in module modelzoo.vision.pytorch.unet.input.scripts.resize_inria_files) (in module modelzoo.vision.pytorch.unet.input.scripts.resize_skm_tea_files) get_preprocess_transform() (in module modelzoo.vision.pytorch.input.preprocessing) get_raw_sequence_lengths() (in module modelzoo.transformers.pytorch.t5.input.utils) get_sampler() (in module modelzoo.vision.pytorch.dit.samplers.get_sampler) get_scale() (cerebras_pytorch.amp.GradScaler method) get_short_documents() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.filter) get_single_item() (modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor.BertCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor.BertCSVDynamicMaskDataProcessor method) (modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor.T5DynamicDataProcessor method) (modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor.TransformerDynamicDataProcessor method) get_streaming_batch_size() (in module modelzoo.common.pytorch.input_utils) get_tokenizer_fns() (in module modelzoo.transformers.data_processing.qa.write_csv_qa) get_tokens_and_labels() (in module modelzoo.transformers.data_processing.bert.ner_data_processor) get_train_dataloader() (in module modelzoo.fc_mnist.pytorch.data) get_urls_for_tokenizer_files() (in module modelzoo.transformers.data_processing.scripts.pile.download) get_urls_from_split() (in module modelzoo.transformers.data_processing.scripts.pile.download) get_verification_args() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) get_vocab() (in module modelzoo.transformers.data_processing.utils) get_vocab_size() (in module modelzoo.common.pytorch.model_utils.vocab_utils) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method) get_vocab_words() (modelzoo.transformers.data_processing.tokenizers.Tokenization.FullTokenizer method) get_whole_word_span() (in module modelzoo.transformers.pytorch.bert.input.utils) get_worker_state() (in module cerebras_pytorch.distributed) global_rate (cerebras_pytorch.utils.benchmark.utils.dataloader.BatchMetrics attribute) (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics attribute) global_sample_rate (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics property) global_step (cerebras_pytorch.utils.benchmark.utils.dataloader.BatchMetrics attribute) glu_bivariate_base_fn() (in module modelzoo.common.pytorch.model_utils.activations) GMPSparsityOptimizer (class in cerebras_pytorch.sparse) GPT2LMHeadModel (class in modelzoo.transformers.pytorch.gpt2.gpt2_model) Gpt2Model (class in modelzoo.transformers.pytorch.gpt2.model) GptHDF5DataProcessor (class in modelzoo.transformers.pytorch.gpt2.input.GptHDF5DataProcessor) GptHDF5MapDataProcessor (class in modelzoo.transformers.pytorch.gpt2.input.GptHDF5MapDataProcessor) GPTJDecoderLayer (class in modelzoo.common.pytorch.layers) GPTJModel (class in modelzoo.transformers.pytorch.gptj.gptj_model) GptjModel (class in modelzoo.transformers.pytorch.gptj.model) GPTLMHeadModelLoss (class in modelzoo.common.pytorch.model_utils.GPTLMHeadModelLoss) GptTextDataProcessor (class in modelzoo.transformers.pytorch.gpt2.input.GptTextDataProcessor) GradScaler (class in cerebras_pytorch.amp) GradScalerParams (class in modelzoo.common.pytorch.run_cstorch_flow) GroupInstanceNorm (class in modelzoo.common.pytorch.layers)

H

H5Reader (class in modelzoo.transformers.data_processing.h5_map_dataset.readers) HalfDType (class in modelzoo.common.pytorch.half_dtype) handle_jsonl() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) Hdf5BaseDataProcessor (class in modelzoo.vision.pytorch.unet.input.Hdf5BaseDataProcessor) Hdf5BaseIterDataProcessor (class in modelzoo.vision.pytorch.unet.input.Hdf5BaseIterDataProcessor) HDF5BasePreprocessor (class in modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor) Hdf5DataProcessor (class in modelzoo.vision.pytorch.unet.input.Hdf5DataProcessor) HDF5Dataset (class in modelzoo.transformers.data_processing.h5_map_dataset.dataset)

HDF5IterableDataProcessor (class in modelzoo.transformers.data_processing.HDF5IterableDataProcessor) HDF5IterableDataset (class in modelzoo.transformers.data_processing.HDF5IterableDataset) HFTokenizer (class in modelzoo.transformers.data_processing.tokenizers.HFTokenizer) hook_module() (cerebras_pytorch.sparse.base.BaseSparsityOptimizer method) HuggingFace_BookCorpus() (in module modelzoo.transformers.data_processing.huggingface.HuggingFace_BookCorpus) HuggingFace_Eli5() (in module modelzoo.transformers.data_processing.huggingface.HuggingFace_Eli5) HuggingFaceDataProcessor (class in modelzoo.transformers.data_processing.huggingface.HuggingFaceDataProcessor) HuggingFaceDataProcessorEli5 (class in modelzoo.transformers.pytorch.gpt2.input.HuggingFaceDataProcessorEli5) HuggingFaceIterableDataProcessorEli5 (class in modelzoo.transformers.pytorch.gpt2.input.HuggingFaceIterableDataProcessorEli5)

I

ImageNetLatentDataset (class in modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor) increment_global_step() (cerebras_pytorch.optim.Optimizer method) increment_last_epoch() (cerebras_pytorch.optim.lr_scheduler.LRScheduler method) (cerebras_pytorch.optim.lr_scheduler.SequentialLR method) index (modelzoo.transformers.data_processing.utils.maskedLmInstance attribute) index() (in module modelzoo.vision.pytorch.dit.layers.GaussianDiffusion) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures method) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction.RawResult method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner.InputFeatures method) init_output_checkpoint() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 class method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 class method)

initialize_hyperparam() (in module cerebras_pytorch.sparse.utils) initialize_sparsity() (cerebras_pytorch.sparse.base.BaseSparsityOptimizer method) input_ids (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) input_mask (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) input_token_ids (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures attribute) InputExample (class in modelzoo.transformers.data_processing.bert.ner_data_processor) InputFeatures (class in modelzoo.transformers.data_processing.qa.qa_utils) (class in modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner) InputGroupScoreShaper (class in cerebras_pytorch.sparse.utils) InriaAerialDataProcessor (class in modelzoo.vision.pytorch.unet.input.InriaAerialDataProcessor) InriaAerialDataset (class in modelzoo.vision.pytorch.unet.input.InriaAerialDataProcessor) insert() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) InverseExponentialTimeDecayLR (class in cerebras_pytorch.optim.lr_scheduler) InverseSquareRootDecayLR (class in cerebras_pytorch.optim.lr_scheduler) is_distributed() (in module modelzoo.transformers.pytorch.input_utils) is_glu_activation() (in module modelzoo.common.pytorch.model_utils.activations) is_gpu_distributed() (in module modelzoo.vision.pytorch.input.utils) is_impossible (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) is_mup_run() (in module modelzoo.common.pytorch.utils) is_partial (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics attribute) items() (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) IterableDatasetSampler (class in modelzoo.common.pytorch.utils) iteration_time (cerebras_pytorch.utils.benchmark.utils.dataloader.EpochMetrics attribute) iterator_creation (cerebras_pytorch.utils.benchmark.utils.dataloader.EpochMetrics attribute)

J

JsonConverter (class in modelzoo.transformers.data_processing.bert.bertsum_data_processor)

K

keys() (modelzoo.transformers.pytorch.bert.input.utils.Vocab method)

L

label (modelzoo.transformers.data_processing.utils.maskedLmInstance attribute) LabelDropout (class in modelzoo.vision.pytorch.dit.input.transforms) labels (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures attribute) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner.InputFeatures attribute) Lamb (class in cerebras_pytorch.optim) LambdaHyperParameter (class in cerebras_pytorch.sparse.utils) LambdaLR (class in cerebras_pytorch.optim.lr_scheduler) lecun_normal_() (in module modelzoo.common.pytorch.model_utils.weight_initializers) lecun_uniform_() (in module modelzoo.common.pytorch.model_utils.weight_initializers) liglu() (in module modelzoo.common.pytorch.model_utils.activations) linear_act() (in module modelzoo.common.pytorch.model_utils.activations) LinearHyperParameter (class in cerebras_pytorch.sparse.utils) LinearLR (class in cerebras_pytorch.optim.lr_scheduler) Lion (class in cerebras_pytorch.optim) ListSchedule (class in cerebras_pytorch.sparse.dynamic) LMDataPreprocessor (class in modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors) load() (in module cerebras_pytorch) load_buffer() (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor.BertSumCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor.BertQADataProcessor method) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor.BertTokenClassifierDataProcessor method) (modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor.BertCSVDataProcessor method) (modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor.BertCSVDynamicMaskDataProcessor method) (modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor.T5DynamicDataProcessor method) (modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor.TransformerDynamicDataProcessor method)

load_state_dict() (cerebras_pytorch.amp.GradScaler method) (cerebras_pytorch.optim.Adam method) (cerebras_pytorch.optim.AdamW method) (cerebras_pytorch.optim.lr_scheduler.ChainedScheduler method) (cerebras_pytorch.optim.lr_scheduler.LRScheduler method) (cerebras_pytorch.optim.lr_scheduler.SequentialLR method) (cerebras_pytorch.optim.Optimizer method) (cerebras_pytorch.utils.data.DataLoader method) (cerebras_pytorch.utils.data.RestartableDataLoader method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.RestartableDataLoader method) local_rate (cerebras_pytorch.utils.benchmark.utils.dataloader.BatchMetrics attribute) log_input_summary() (in module modelzoo.common.pytorch.run_cstorch_flow) LRScheduler (class in cerebras_pytorch.optim.lr_scheduler) lsh() (in module modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicate_pairs)

M

main() (in module modelzoo.common.pytorch.run_utils) (in module modelzoo.fc_mnist.pytorch.run) (in module modelzoo.transformers.data_processing.h5_map_dataset.preprocess_pile) (in module modelzoo.transformers.data_processing.huggingface.HF_converter_example_BookCorpus) (in module modelzoo.transformers.data_processing.huggingface.HF_converter_example_Eli5) (in module modelzoo.transformers.data_processing.qa.write_csv_qa) (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.create_hdf5_dataset) (in module modelzoo.transformers.data_processing.scripts.pile.download) (in module modelzoo.transformers.data_processing.slimpajama.main) (in module modelzoo.transformers.data_processing.slimpajama.split_dataset) (in module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.run) (in module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.run) (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.run) (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction) (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.run) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only_static_masking) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_static_masking) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_hdf5_files) (in module modelzoo.transformers.pytorch.bert.run) (in module modelzoo.transformers.pytorch.bloom.run) (in module modelzoo.transformers.pytorch.falcon.run) (in module modelzoo.transformers.pytorch.gpt2.run) (in module modelzoo.transformers.pytorch.gpt3.run) (in module modelzoo.transformers.pytorch.gptj.fine_tuning.continuous_pretraining.data_processing.split_trc_dataset) (in module modelzoo.transformers.pytorch.gptj.run) (in module modelzoo.transformers.pytorch.layers_api_demo.run) (in module modelzoo.transformers.pytorch.llama.run) (in module modelzoo.transformers.pytorch.mpt.run) (in module modelzoo.transformers.pytorch.santacoder.run) (in module modelzoo.transformers.pytorch.starcoder.run) (in module modelzoo.transformers.pytorch.t5.run) (in module modelzoo.transformers.pytorch.transformer.input.data_processing.create_meta) (in module modelzoo.transformers.pytorch.transformer.input.data_processing.split_files) (in module modelzoo.transformers.pytorch.transformer.run) (in module modelzoo.vision.pytorch.dit.run) (in module modelzoo.vision.pytorch.unet.input.scripts.create_hdf5_files) (in module modelzoo.vision.pytorch.unet.run) 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) 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 module modelzoo.common module modelzoo.common.input module modelzoo.common.input.utils module modelzoo.common.model_utils module modelzoo.common.model_utils.count_lines module modelzoo.common.model_utils.sparsity module modelzoo.common.pytorch module modelzoo.common.pytorch.dump_context module modelzoo.common.pytorch.half_dtype module modelzoo.common.pytorch.input_utils module modelzoo.common.pytorch.layers module modelzoo.common.pytorch.model_utils module modelzoo.common.pytorch.model_utils.activations module modelzoo.common.pytorch.model_utils.BertPretrainModelLoss module modelzoo.common.pytorch.model_utils.checkpoint_converters module modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter module modelzoo.common.pytorch.model_utils.checkpoint_converters.bert module modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune module modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs module modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs module modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon module modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b module modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b module modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs module modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs module modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs module modelzoo.common.pytorch.model_utils.checkpoint_converters.helper module modelzoo.common.pytorch.model_utils.checkpoint_converters.llama module modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt module modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs module modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta module modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs module modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder module modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder module modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints module modelzoo.common.pytorch.model_utils.checkpoint_converters.t5 module modelzoo.common.pytorch.model_utils.checkpoint_converters.vit module modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae module modelzoo.common.pytorch.model_utils.convert_checkpoint module modelzoo.common.pytorch.model_utils.convert_config_to_mup module modelzoo.common.pytorch.model_utils.create_initializer module modelzoo.common.pytorch.model_utils.GPTLMHeadModelLoss module modelzoo.common.pytorch.model_utils.norms module modelzoo.common.pytorch.model_utils.RotaryPositionEmbeddingHelper module modelzoo.common.pytorch.model_utils.T5ForConditionalGenerationLoss module modelzoo.common.pytorch.model_utils.vocab_utils module modelzoo.common.pytorch.model_utils.weight_initializers module modelzoo.common.pytorch.run_cstorch_flow module modelzoo.common.pytorch.run_utils module modelzoo.common.pytorch.utils module modelzoo.common.run_utils module modelzoo.common.run_utils.cli_parser module modelzoo.common.run_utils.cli_pytorch module modelzoo.common.run_utils.utils module modelzoo.fc_mnist module modelzoo.fc_mnist.pytorch module modelzoo.fc_mnist.pytorch.data module modelzoo.fc_mnist.pytorch.model module modelzoo.fc_mnist.pytorch.prepare_data module modelzoo.fc_mnist.pytorch.run module modelzoo.fc_mnist.pytorch.utils module modelzoo.transformers module modelzoo.transformers.data_processing module modelzoo.transformers.data_processing.bert module modelzoo.transformers.data_processing.bert.bertsum_data_processor module modelzoo.transformers.data_processing.bert.dynamic_processor module modelzoo.transformers.data_processing.bert.mlm_only_processor module modelzoo.transformers.data_processing.bert.ner_data_processor module modelzoo.transformers.data_processing.bert.sentence_pair_processor module modelzoo.transformers.data_processing.GenericDataProcessor module modelzoo.transformers.data_processing.h5_map_dataset module modelzoo.transformers.data_processing.h5_map_dataset.dataset module modelzoo.transformers.data_processing.h5_map_dataset.preprocess_pile module modelzoo.transformers.data_processing.h5_map_dataset.readers module modelzoo.transformers.data_processing.h5_map_dataset.samplers module modelzoo.transformers.data_processing.HDF5IterableDataProcessor module modelzoo.transformers.data_processing.HDF5IterableDataset module modelzoo.transformers.data_processing.huggingface module modelzoo.transformers.data_processing.huggingface.CSDataCollatorForLanguageModeling module modelzoo.transformers.data_processing.huggingface.HF_converter_example_BookCorpus module modelzoo.transformers.data_processing.huggingface.HF_converter_example_Eli5 module modelzoo.transformers.data_processing.huggingface.HuggingFace_BookCorpus module modelzoo.transformers.data_processing.huggingface.HuggingFace_Eli5 module modelzoo.transformers.data_processing.huggingface.HuggingFaceDataProcessor module modelzoo.transformers.data_processing.qa module modelzoo.transformers.data_processing.qa.qa_utils module modelzoo.transformers.data_processing.qa.write_csv_qa module modelzoo.transformers.data_processing.scripts module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.convert_dataset_to_HDF5 module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.create_hdf5_dataset module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils module modelzoo.transformers.data_processing.scripts.hdf5_shuffling module modelzoo.transformers.data_processing.scripts.hdf5_shuffling.h5_dataset_shuffle module modelzoo.transformers.data_processing.scripts.pile module modelzoo.transformers.data_processing.scripts.pile.download module modelzoo.transformers.data_processing.scripts.pubmed module modelzoo.transformers.data_processing.scripts.pubmed.preprocess module modelzoo.transformers.data_processing.scripts.pubmed.preprocess.Downloader module modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextFormatting module modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextSharding module modelzoo.transformers.data_processing.slimpajama module modelzoo.transformers.data_processing.slimpajama.dedup module modelzoo.transformers.data_processing.slimpajama.dedup.dedup_train module modelzoo.transformers.data_processing.slimpajama.dedup.generate_connected_components module modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicate_pairs module modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicates_dict module modelzoo.transformers.data_processing.slimpajama.dedup.to_hash module modelzoo.transformers.data_processing.slimpajama.main module modelzoo.transformers.data_processing.slimpajama.preprocessing module modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets module modelzoo.transformers.data_processing.slimpajama.preprocessing.filter module modelzoo.transformers.data_processing.slimpajama.preprocessing.normalize_text module modelzoo.transformers.data_processing.slimpajama.preprocessing.shuffle_holdout module modelzoo.transformers.data_processing.slimpajama.split_dataset module modelzoo.transformers.data_processing.slimpajama.utils module modelzoo.transformers.data_processing.slimpajama.utils.utils module modelzoo.transformers.data_processing.tokenizers module modelzoo.transformers.data_processing.tokenizers.BPETokenizer module modelzoo.transformers.data_processing.tokenizers.HFTokenizer module modelzoo.transformers.data_processing.tokenizers.Tokenization module modelzoo.transformers.data_processing.utils module modelzoo.transformers.pytorch module modelzoo.transformers.pytorch.bert module modelzoo.transformers.pytorch.bert.bert_finetune_models module modelzoo.transformers.pytorch.bert.bert_model module modelzoo.transformers.pytorch.bert.bert_pretrain_models module modelzoo.transformers.pytorch.bert.data module modelzoo.transformers.pytorch.bert.fine_tuning module modelzoo.transformers.pytorch.bert.fine_tuning.classifier module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.data module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.model module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.run module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.utils module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.data module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.BertSumCSVDataProcessor module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.write_csv_data module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.model module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.run module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.utils module modelzoo.transformers.pytorch.bert.fine_tuning.qa module modelzoo.transformers.pytorch.bert.fine_tuning.qa.data module modelzoo.transformers.pytorch.bert.fine_tuning.qa.input module modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor module modelzoo.transformers.pytorch.bert.fine_tuning.qa.model module modelzoo.transformers.pytorch.bert.fine_tuning.qa.run module modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction module modelzoo.transformers.pytorch.bert.fine_tuning.qa.utils module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.data module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.model module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.run module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.utils module modelzoo.transformers.pytorch.bert.input module modelzoo.transformers.pytorch.bert.input.BertCSVDataProcessor module modelzoo.transformers.pytorch.bert.input.BertCSVDynamicMaskDataProcessor module modelzoo.transformers.pytorch.bert.input.scripts module modelzoo.transformers.pytorch.bert.input.scripts.create_csv module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only_static_masking module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_static_masking module modelzoo.transformers.pytorch.bert.input.scripts.create_hdf5_files module modelzoo.transformers.pytorch.bert.input.scripts.parser_utils module modelzoo.transformers.pytorch.bert.input.utils module modelzoo.transformers.pytorch.bert.model module modelzoo.transformers.pytorch.bert.run module modelzoo.transformers.pytorch.bert.utils module modelzoo.transformers.pytorch.bloom module modelzoo.transformers.pytorch.bloom.run module modelzoo.transformers.pytorch.falcon module modelzoo.transformers.pytorch.falcon.run module modelzoo.transformers.pytorch.gpt2 module modelzoo.transformers.pytorch.gpt2.data module modelzoo.transformers.pytorch.gpt2.gpt2_model module modelzoo.transformers.pytorch.gpt2.input module modelzoo.transformers.pytorch.gpt2.input.DummyDataProcessor module modelzoo.transformers.pytorch.gpt2.input.DummyIterableDataProcessor module modelzoo.transformers.pytorch.gpt2.input.GptHDF5DataProcessor module modelzoo.transformers.pytorch.gpt2.input.GptHDF5MapDataProcessor module modelzoo.transformers.pytorch.gpt2.input.GptTextDataProcessor module modelzoo.transformers.pytorch.gpt2.input.HuggingFaceDataProcessorEli5 module modelzoo.transformers.pytorch.gpt2.input.HuggingFaceIterableDataProcessorEli5 module modelzoo.transformers.pytorch.gpt2.input.scripts module modelzoo.transformers.pytorch.gpt2.input.scripts.data_processor_utils module modelzoo.transformers.pytorch.gpt2.model module modelzoo.transformers.pytorch.gpt2.run module modelzoo.transformers.pytorch.gpt2.scripts module modelzoo.transformers.pytorch.gpt2.scripts.fold_mup module modelzoo.transformers.pytorch.gpt2.sparse_mask module modelzoo.transformers.pytorch.gpt2.utils module modelzoo.transformers.pytorch.gpt3 module modelzoo.transformers.pytorch.gpt3.run module modelzoo.transformers.pytorch.gptj module modelzoo.transformers.pytorch.gptj.fine_tuning module modelzoo.transformers.pytorch.gptj.fine_tuning.continuous_pretraining module modelzoo.transformers.pytorch.gptj.fine_tuning.continuous_pretraining.data_processing module modelzoo.transformers.pytorch.gptj.fine_tuning.continuous_pretraining.data_processing.split_trc_dataset module modelzoo.transformers.pytorch.gptj.gptj_model module modelzoo.transformers.pytorch.gptj.model module modelzoo.transformers.pytorch.gptj.run module modelzoo.transformers.pytorch.gptj.utils module modelzoo.transformers.pytorch.input_utils module modelzoo.transformers.pytorch.layers_api_demo module modelzoo.transformers.pytorch.layers_api_demo.cb_transformer module modelzoo.transformers.pytorch.layers_api_demo.data module modelzoo.transformers.pytorch.layers_api_demo.model module modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer module modelzoo.transformers.pytorch.layers_api_demo.run module modelzoo.transformers.pytorch.llama module modelzoo.transformers.pytorch.llama.run module modelzoo.transformers.pytorch.mpt module modelzoo.transformers.pytorch.mpt.run module modelzoo.transformers.pytorch.santacoder module modelzoo.transformers.pytorch.santacoder.run module modelzoo.transformers.pytorch.starcoder module modelzoo.transformers.pytorch.starcoder.run module modelzoo.transformers.pytorch.t5 module modelzoo.transformers.pytorch.t5.data module modelzoo.transformers.pytorch.t5.input module modelzoo.transformers.pytorch.t5.input.data_processing module

modelzoo.transformers.pytorch.t5.input.data_processing.preprocess_t5_dataset module modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor module modelzoo.transformers.pytorch.t5.input.utils module modelzoo.transformers.pytorch.t5.model module modelzoo.transformers.pytorch.t5.run module modelzoo.transformers.pytorch.t5.t5_model module modelzoo.transformers.pytorch.t5.utils module modelzoo.transformers.pytorch.transformer module modelzoo.transformers.pytorch.transformer.data module modelzoo.transformers.pytorch.transformer.input module modelzoo.transformers.pytorch.transformer.input.data_processing module modelzoo.transformers.pytorch.transformer.input.data_processing.create_meta module modelzoo.transformers.pytorch.transformer.input.data_processing.split_files module modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor module modelzoo.transformers.pytorch.transformer.run module modelzoo.transformers.pytorch.transformer.utils module modelzoo.transformers.pytorch.transformer_utils module modelzoo.transformers.utils module modelzoo.transformers.utils.attention_utils module modelzoo.vision module modelzoo.vision.pytorch module modelzoo.vision.pytorch.dit module modelzoo.vision.pytorch.dit.checkpoint_converter module modelzoo.vision.pytorch.dit.data module modelzoo.vision.pytorch.dit.display_images module modelzoo.vision.pytorch.dit.input module modelzoo.vision.pytorch.dit.input.DiffusionBaseProcessor module modelzoo.vision.pytorch.dit.input.DiffusionLatentImageNet1KProcessor module modelzoo.vision.pytorch.dit.input.scripts module modelzoo.vision.pytorch.dit.input.transforms module modelzoo.vision.pytorch.dit.layers module modelzoo.vision.pytorch.dit.layers.DiTDecoder module modelzoo.vision.pytorch.dit.layers.DiTDecoderLayer module modelzoo.vision.pytorch.dit.layers.GaussianDiffusion module modelzoo.vision.pytorch.dit.layers.RegressionHead module modelzoo.vision.pytorch.dit.layers.schedulers module modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer module modelzoo.vision.pytorch.dit.layers.vae module modelzoo.vision.pytorch.dit.layers.vae.DownEncoderBlock2D module modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D module modelzoo.vision.pytorch.dit.layers.vae.UNetMidBlock2D module modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D module modelzoo.vision.pytorch.dit.layers.vae.utils module modelzoo.vision.pytorch.dit.layers.vae.VAEDecoder module modelzoo.vision.pytorch.dit.layers.vae.VAEEncoder module modelzoo.vision.pytorch.dit.layers.vae.VAEModel module modelzoo.vision.pytorch.dit.model module modelzoo.vision.pytorch.dit.modeling_dit module modelzoo.vision.pytorch.dit.pipeline module modelzoo.vision.pytorch.dit.run module modelzoo.vision.pytorch.dit.sample_generator module modelzoo.vision.pytorch.dit.sample_generator_dit module modelzoo.vision.pytorch.dit.samplers module modelzoo.vision.pytorch.dit.samplers.DDIMSampler module modelzoo.vision.pytorch.dit.samplers.DDPMSampler module modelzoo.vision.pytorch.dit.samplers.get_sampler module modelzoo.vision.pytorch.dit.samplers.sampler_utils module modelzoo.vision.pytorch.dit.samplers.SamplerBase module modelzoo.vision.pytorch.dit.utils module modelzoo.vision.pytorch.input module modelzoo.vision.pytorch.input.classification module modelzoo.vision.pytorch.input.classification.dataset_factory module modelzoo.vision.pytorch.input.classification.mixup module modelzoo.vision.pytorch.input.classification.sampler module modelzoo.vision.pytorch.input.classification.utils module modelzoo.vision.pytorch.input.preprocessing module modelzoo.vision.pytorch.input.transforms module modelzoo.vision.pytorch.input.utils module modelzoo.vision.pytorch.unet module modelzoo.vision.pytorch.unet.data module modelzoo.vision.pytorch.unet.input module modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor module modelzoo.vision.pytorch.unet.input.Hdf5BaseDataProcessor module modelzoo.vision.pytorch.unet.input.Hdf5BaseIterDataProcessor module modelzoo.vision.pytorch.unet.input.Hdf5DataProcessor module modelzoo.vision.pytorch.unet.input.InriaAerialDataProcessor module modelzoo.vision.pytorch.unet.input.preprocessing_utils module modelzoo.vision.pytorch.unet.input.scripts module modelzoo.vision.pytorch.unet.input.scripts.create_hdf5_files module modelzoo.vision.pytorch.unet.input.scripts.resize_inria_files module modelzoo.vision.pytorch.unet.input.scripts.resize_skm_tea_files module modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor module modelzoo.vision.pytorch.unet.input.SkmDataProcessor module modelzoo.vision.pytorch.unet.input.UNetDataProcessor module modelzoo.vision.pytorch.unet.layers module modelzoo.vision.pytorch.unet.layers.Decoder module modelzoo.vision.pytorch.unet.layers.Encoder module modelzoo.vision.pytorch.unet.layers.UNetBlock module modelzoo.vision.pytorch.unet.model module modelzoo.vision.pytorch.unet.modeling_unet module modelzoo.vision.pytorch.unet.run module modelzoo.vision.pytorch.unet.utils module modelzoo.vision.pytorch.utils module modelzoo.vision.pytorch.utils.run_utils module module cerebras_pytorch.sparse.base cerebras_pytorch.sparse.dynamic cerebras_pytorch.sparse.init cerebras_pytorch.sparse.utils modelzoo modelzoo.common modelzoo.common.input modelzoo.common.input.utils modelzoo.common.model_utils modelzoo.common.model_utils.count_lines modelzoo.common.model_utils.sparsity modelzoo.common.pytorch modelzoo.common.pytorch.dump_context modelzoo.common.pytorch.half_dtype 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.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_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.mpt modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs 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.checkpoint_converters.vit modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae 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.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_flow 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.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.qa modelzoo.transformers.pytorch.bert.fine_tuning.qa.data modelzoo.transformers.pytorch.bert.fine_tuning.qa.input modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor modelzoo.transformers.pytorch.bert.fine_tuning.qa.model modelzoo.transformers.pytorch.bert.fine_tuning.qa.run modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction modelzoo.transformers.pytorch.bert.fine_tuning.qa.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.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.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.scripts.fold_mup 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.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.T5DynamicDataProcessor 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.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.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.dataset_factory modelzoo.vision.pytorch.input.classification.mixup modelzoo.vision.pytorch.input.classification.sampler modelzoo.vision.pytorch.input.classification.utils modelzoo.vision.pytorch.input.preprocessing modelzoo.vision.pytorch.input.transforms modelzoo.vision.pytorch.input.utils modelzoo.vision.pytorch.unet modelzoo.vision.pytorch.unet.data modelzoo.vision.pytorch.unet.input modelzoo.vision.pytorch.unet.input.CityscapesDataProcessor modelzoo.vision.pytorch.unet.input.Hdf5BaseDataProcessor modelzoo.vision.pytorch.unet.input.Hdf5BaseIterDataProcessor modelzoo.vision.pytorch.unet.input.Hdf5DataProcessor modelzoo.vision.pytorch.unet.input.InriaAerialDataProcessor modelzoo.vision.pytorch.unet.input.preprocessing_utils modelzoo.vision.pytorch.unet.input.scripts modelzoo.vision.pytorch.unet.input.scripts.create_hdf5_files modelzoo.vision.pytorch.unet.input.scripts.resize_inria_files modelzoo.vision.pytorch.unet.input.scripts.resize_skm_tea_files modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor modelzoo.vision.pytorch.unet.input.SkmDataProcessor modelzoo.vision.pytorch.unet.input.UNetDataProcessor modelzoo.vision.pytorch.unet.layers modelzoo.vision.pytorch.unet.layers.Decoder modelzoo.vision.pytorch.unet.layers.Encoder modelzoo.vision.pytorch.unet.layers.UNetBlock modelzoo.vision.pytorch.unet.model modelzoo.vision.pytorch.unet.modeling_unet modelzoo.vision.pytorch.unet.run modelzoo.vision.pytorch.unet.utils modelzoo.vision.pytorch.utils modelzoo.vision.pytorch.utils.run_utils 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)

N

NAdam (class in cerebras_pytorch.optim) name() (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaBooksDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaC4Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaCommonCrawlDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaGithubDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaReplication method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaStackExchangeDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaWikipediaDataset method) NERProcessor (class in modelzoo.transformers.data_processing.bert.ner_data_processor)

next() (modelzoo.common.pytorch.utils.SampleGenerator method) NLGPreprocessor (class in modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor) NLTKSegmenter (class in modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextSharding) noise_token_span_to_unique_sentinel() (in module modelzoo.transformers.pytorch.t5.input.utils) NoiseGenerator (class in modelzoo.vision.pytorch.dit.input.transforms) normalize_tensor_transform() (in module modelzoo.vision.pytorch.unet.input.preprocessing_utils) normalize_text() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.normalize_text) num_docs() (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaReplication method) num_tasks() (in module modelzoo.transformers.pytorch.input_utils) (in module modelzoo.vision.pytorch.input.utils) num_updates (cerebras_pytorch.metrics.Metric property)

O

OneCycleLR (class in cerebras_pytorch.optim.lr_scheduler) ones() (in module cerebras_pytorch) ones_like() (in module cerebras_pytorch) Optimizer (class in cerebras_pytorch.optim)

optimizer_step() (in module cerebras_pytorch.amp) optimizer_step_with_summaries() (in module modelzoo.common.pytorch.run_cstorch_flow) output_results() (in module modelzoo.transformers.data_processing.slimpajama.dedup.to_hash) OutputGroupScoreShaper (class in cerebras_pytorch.sparse.utils)

P

pad_input_sequence() (in module modelzoo.transformers.data_processing.utils) pad_instance_to_max_seq_length() (in module modelzoo.transformers.data_processing.utils) pad_t5_input_features() (in module modelzoo.transformers.pytorch.t5.input.utils) parse_args() (in module modelzoo.transformers.data_processing.h5_map_dataset.preprocess_pile) (in module modelzoo.transformers.data_processing.qa.write_csv_qa) (in module modelzoo.transformers.data_processing.scripts.pile.download) (in module modelzoo.transformers.data_processing.scripts.pubmed.preprocess.Downloader) (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.normalize_text) (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_mlm_only_static_masking) (in module modelzoo.transformers.pytorch.bert.input.scripts.create_csv_static_masking) parse_argument() (in module modelzoo.common.pytorch.model_utils.convert_config_to_mup) parse_ner_row() (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.BertTokenClassifierDataProcessor) parse_qa_row() (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.input.BertQADataProcessor) parse_text() (in module modelzoo.transformers.pytorch.bert.input.utils) (in module modelzoo.transformers.pytorch.t5.input.utils) pass_1_shuffle() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.shuffle_holdout) pass_2_shuffle_holdout() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.shuffle_holdout) PerplexityMetric (class in cerebras_pytorch.metrics) PiecewiseConstantLR (class in cerebras_pytorch.optim.lr_scheduler) PolynomialLR (class in cerebras_pytorch.optim.lr_scheduler) pop() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) popitem() (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) PositionalEncoding (class in modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer) post_checkpoint_convert() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 method) post_model_convert() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 method)

post_process_params() (in module modelzoo.common.run_utils.cli_parser) PowerHyperParameter (class in cerebras_pytorch.sparse.utils) pre_checkpoint_convert() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 method) pre_model_convert() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 method) predict_input_dataloader() (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.data) preinitialize() (cerebras_pytorch.optim.Adadelta method) (cerebras_pytorch.optim.Adafactor method) (cerebras_pytorch.optim.Adagrad method) (cerebras_pytorch.optim.Adamax method) (cerebras_pytorch.optim.ASGD method) (cerebras_pytorch.optim.Lamb method) (cerebras_pytorch.optim.Lion method) (cerebras_pytorch.optim.NAdam method) (cerebras_pytorch.optim.Optimizer method) (cerebras_pytorch.optim.RAdam method) (cerebras_pytorch.optim.RMSprop method) (cerebras_pytorch.optim.Rprop method) (cerebras_pytorch.optim.SGD method) preprocessing_generator() (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method) PreprocessInstance (class in modelzoo.transformers.data_processing.bert.dynamic_processor) previous_timestep() (modelzoo.vision.pytorch.dit.samplers.DDIMSampler.DDIMSampler method) (modelzoo.vision.pytorch.dit.samplers.DDPMSampler.DDPMSampler method) (modelzoo.vision.pytorch.dit.samplers.SamplerBase.SamplerBase method) process_dataset() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) Processor (class in modelzoo.vision.pytorch.input.classification.dataset_factory) profile_activities (cerebras_pytorch.utils.benchmark.utils.dataloader.BatchMetrics attribute)

Q

quick_gelu() (in module modelzoo.common.pytorch.model_utils.activations)

R

RAdam (class in cerebras_pytorch.optim) randn_tensor() (in module modelzoo.vision.pytorch.dit.layers.vae.utils) random() (in module cerebras_pytorch.sparse.init) random_spans_noise_mask() (in module modelzoo.transformers.pytorch.t5.input.utils) RandomCutmix (class in modelzoo.vision.pytorch.input.classification.mixup) RandomMixup (class in modelzoo.vision.pytorch.input.classification.mixup) RandomScoreTiebreaker (class in cerebras_pytorch.sparse.utils) RawResult (class in modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction) read_checkpoint() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) read_params_file() (in module modelzoo.common.run_utils.cli_parser) read_scalar() (cerebras_pytorch.utils.tensorboard.SummaryReader method) read_squad_examples() (in module modelzoo.transformers.data_processing.qa.qa_utils) read_tensor() (cerebras_pytorch.utils.tensorboard.SummaryReader method) Reader (class in modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) recreate_dataset() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.normalize_text) RedPajamaArXivDataset (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) RedPajamaBooksDataset (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) RedPajamaC4Dataset (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) RedPajamaCommonCrawlDataset (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) RedPajamaGithubDataset (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) RedPajamaReplication (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) RedPajamaStackExchangeDataset (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) RedPajamaWikipediaDataset (class in modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) redpj_datasets() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets) register_state() (cerebras_pytorch.metrics.Metric method) registry (cerebras_pytorch.metrics.Metric attribute) reglu() (in module modelzoo.common.pytorch.model_utils.activations) RegressionHead (class in modelzoo.vision.pytorch.dit.layers.RegressionHead) relative_position_bucket() (modelzoo.common.pytorch.layers.RelativePositionEmbeddingLayer static method) RelativePositionEmbeddingLayer (class in modelzoo.common.pytorch.layers) reload() (cerebras_pytorch.utils.tensorboard.SummaryReader method) remove() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) RepeatedAugSampler (class in modelzoo.vision.pytorch.input.classification.sampler) replaceKey() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_CS_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseCheckpointConverter_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter_CS_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseConfigConverter_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.BaseDictionaryConverter static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS16_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS16_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_CS17_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.ConfigConverter_Bert_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_Bert_CS17_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertLayerNorm_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_CS16_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_CS16_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert.Converter_BertPretrainModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForQuestionAnswering_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForQuestionAnswering_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForSequenceClassification_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForSequenceClassification_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForTokenClassification_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.ConfigConverter_BertForTokenClassification_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertFinetuneModel_CS16_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForQuestionAnswering_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForSequenceClassification_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bert_finetune.Converter_BertForTokenClassification_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomLMHeadModel_CS19_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.ConfigConverter_BloomModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_Bloom_Attention_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_CS19_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomLMHeadModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.bloom_hf_cs.Converter_BloomModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.ConfigConverter_BTLMModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMLMHeadModel_WithoutModelPrefix_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.btlm_hf_cs.Converter_BTLMModel_WithoutModelPrefix_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.ConfigConverter_Falcon_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_Headless_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon.Converter_Falcon_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.ConfigConverter_Falcon_40B_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Attention_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_Headless_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_40b.Converter_Falcon_40B_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.ConfigConverter_Falcon_7B_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Attention_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_Headless_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.falcon_7b.Converter_Falcon_7B_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.ConfigConverter_GPT2Model_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2_Attention_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2LMHeadModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs.Converter_GPT2Model_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_Headless_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.ConfigConverter_GPT_Neox_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Attention_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_Headless_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt_neox_hf_cs.Converter_GPT_Neox_LMHeadModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.ConfigConverter_GPTJModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Attention_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_Headless_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.gptj_hf_cs.Converter_GPTJ_LMHeadModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LLaMa_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LLaMa_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.ConfigConverter_LlamaModel_CS19_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaAttention_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_CS19_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaForCausalLM_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.llama.Converter_LlamaModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.ConfigConverter_MPT_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTAttention_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTForCausalLM_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.mpt.Converter_MPTModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.ConfigConverter_OPTModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Attention_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_Headless_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.opt_hf_cs.Converter_OPT_LMHeadModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.ConfigConverter_Roberta_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.roberta.Converter_RobertaPretrainModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.ConfigConverter_Codegen_Model_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Attention_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_Headless_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.salesforce_codegen_hf_cs.Converter_Codegen_LMHeadModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.ConfigConverter_SantacoderModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_Santacoder_Attention_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderLMHeadModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.santacoder.Converter_SantacoderModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.ConfigConverter_StarcoderModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_Starcoder_Attention_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderForCausalLM_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.starcoder.Converter_StarcoderModel_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS16_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS16_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS17_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.ConfigConverter_T5_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS16_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS17_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_CS18_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS17 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS18 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.t5.Converter_T5_HF_CS20 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.ConfigConverter_ViT_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Core_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_Headless_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit.Converter_ViT_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.ConfigConverter_ViTMAE_HF_CS19 static method) (modelzoo.common.pytorch.model_utils.checkpoint_converters.vit_mae.Converter_ViTMAE_Core_HF_CS19 static method)

rescale_zero_terminal_snr() (in module modelzoo.vision.pytorch.dit.samplers.sampler_utils) reset() (cerebras_pytorch.metrics.AccuracyMetric method) (cerebras_pytorch.metrics.DiceCoefficientMetric method) (cerebras_pytorch.metrics.FBetaScoreMetric method) (cerebras_pytorch.metrics.MeanIOUMetric method) (cerebras_pytorch.metrics.Metric method) (cerebras_pytorch.metrics.PerplexityMetric method) resize_image_with_crop_or_pad_transform() (in module modelzoo.vision.pytorch.unet.input.preprocessing_utils) ResnetBlock2D (class in modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D) RestartableDataLoader (class in cerebras_pytorch.utils.data) (class in modelzoo.transformers.data_processing.h5_map_dataset.dataset) (class in modelzoo.transformers.data_processing.HDF5IterableDataset) reverse() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) RigLSparsityOptimizer (class in cerebras_pytorch.sparse) rm_if_exists() (in module modelzoo.transformers.data_processing.slimpajama.utils.utils) RMSNorm (class in modelzoo.common.pytorch.layers) RMSprop (class in cerebras_pytorch.optim) RotaryPositionEmbeddingHelper (class in modelzoo.common.pytorch.model_utils.RotaryPositionEmbeddingHelper) rotation_90_transform() (in module modelzoo.vision.pytorch.unet.input.preprocessing_utils) RougeBasedLabelsFormatter (class in modelzoo.transformers.data_processing.bert.bertsum_data_processor) Rprop (class in cerebras_pytorch.optim) run() (in module modelzoo.common.pytorch.run_utils) (modelzoo.vision.pytorch.dit.sample_generator.SampleGenerator method) (modelzoo.vision.pytorch.dit.sample_generator_dit.DiTSampleGenerator method) run_cstorch_eval() (in module modelzoo.common.pytorch.run_cstorch_flow) run_cstorch_flow() (in module modelzoo.common.pytorch.run_cstorch_flow) run_cstorch_train() (in module modelzoo.common.pytorch.run_cstorch_flow) run_with_params() (in module modelzoo.common.pytorch.run_utils) RunConfigParamsValidator (class in modelzoo.common.pytorch.utils)

S

SampleGenerator (class in modelzoo.common.pytorch.utils) (class in modelzoo.vision.pytorch.dit.sample_generator) SamplerBase (class in modelzoo.vision.pytorch.dit.samplers.SamplerBase) sampling_time_ns (cerebras_pytorch.utils.benchmark.utils.dataloader.BatchMetrics attribute) save() (in module cerebras_pytorch) save_params() (in module modelzoo.common.input.utils) save_run_info() (in module modelzoo.transformers.data_processing.h5_map_dataset.preprocess_pile) scalar_names() (cerebras_pytorch.utils.tensorboard.SummaryReader method) scale() (cerebras_pytorch.amp.GradScaler method) ScoreFlattener (class in cerebras_pytorch.sparse.utils) ScoreShaper (class in cerebras_pytorch.sparse.utils) ScoreTiebreaker (class in cerebras_pytorch.sparse.utils) seed_runs() (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method) segment_ids (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) (modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.input.utils.BertInputFeatures attribute) select_patches() (modelzoo.common.pytorch.layers.ViTEmbeddingLayer method) select_random_chunk() (in module modelzoo.transformers.pytorch.t5.input.utils) SentencePairInstance (class in modelzoo.transformers.data_processing.bert.sentence_pair_processor) SequentialLR (class in cerebras_pytorch.optim.lr_scheduler) serialize_state_dict() (cerebras_pytorch.utils.data.DataLoader method) set_attention_kernel() (in module modelzoo.transformers.pytorch.gpt2.utils) set_defaults() (in module modelzoo.fc_mnist.pytorch.utils) (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) (in module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.utils) (in module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.utils) (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.utils) (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.utils) (in module modelzoo.transformers.pytorch.bert.utils) (in module modelzoo.transformers.pytorch.gpt2.utils) (in module modelzoo.transformers.pytorch.gptj.utils) (in module modelzoo.transformers.pytorch.t5.utils) (in module modelzoo.transformers.pytorch.transformer.utils) (in module modelzoo.vision.pytorch.dit.utils) (in module modelzoo.vision.pytorch.unet.utils) set_lr_lambda() (cerebras_pytorch.optim.lr_scheduler.LambdaLR method) (cerebras_pytorch.optim.lr_scheduler.MultiplicativeLR method) set_param_group_hyperparam() (in module cerebras_pytorch.sparse.utils) set_sampling_timesteps() (in module modelzoo.vision.pytorch.dit.samplers.sampler_utils) set_scale_fn() (cerebras_pytorch.optim.lr_scheduler.CyclicLR method) set_state() (modelzoo.transformers.data_processing.h5_map_dataset.samplers.CBSampler method) (modelzoo.transformers.data_processing.HDF5IterableDataset.HDF5IterableDataset method) set_timesteps() (modelzoo.vision.pytorch.dit.samplers.DDIMSampler.DDIMSampler method) (modelzoo.vision.pytorch.dit.samplers.DDPMSampler.DDPMSampler method) (modelzoo.vision.pytorch.dit.samplers.SamplerBase.SamplerBase method) set_transforms() (modelzoo.vision.pytorch.input.classification.dataset_factory.VisionSubset method) setdefault() (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) SETSparsityOptimizer (class in cerebras_pytorch.sparse) setup_hooks() (modelzoo.common.pytorch.dump_context.DumpContext method) setup_logging() (in module modelzoo.common.pytorch.utils) SeverstalBinaryClassDataProcessor (class in modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor) SeverstalBinaryClassDataset (class in modelzoo.vision.pytorch.unet.input.SeverstalBinaryClassDataProcessor) SGD (class in cerebras_pytorch.optim) sha256str() (in module modelzoo.transformers.data_processing.slimpajama.utils.utils) shard_and_shuffle_data() (in module modelzoo.transformers.pytorch.bert.input.utils) shard_list_contiguous() (in module modelzoo.transformers.pytorch.input_utils) shard_list_interleaved() (in module modelzoo.transformers.pytorch.input_utils) shard_list_of_chunks_contiguous() (in module modelzoo.transformers.pytorch.input_utils) ShardedSampler (class in modelzoo.transformers.pytorch.input_utils) (class in modelzoo.vision.pytorch.input.utils) Sharder (class in modelzoo.transformers.data_processing.h5_map_dataset.samplers) Sharding (class in modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextSharding) short_documents_path() (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaBooksDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaC4Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaCommonCrawlDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaGithubDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaReplication method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaStackExchangeDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaWikipediaDataset method) shuffle() (in module modelzoo.transformers.pytorch.t5.input.utils) shuffle_dataset() (in module modelzoo.transformers.data_processing.scripts.hdf5_shuffling.h5_dataset_shuffle) sideband_eval_all() (in module modelzoo.common.pytorch.run_utils) sideband_train_eval_all() (in module modelzoo.common.pytorch.run_utils) size() (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaArXivDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaBooksDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaC4Dataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaCommonCrawlDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaGithubDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaReplication method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaStackExchangeDataset method) (modelzoo.transformers.data_processing.slimpajama.preprocessing.datasets.RedPajamaWikipediaDataset method)

SkmDataProcessor (class in modelzoo.vision.pytorch.unet.input.SkmDataProcessor) smooth_loss() (in module modelzoo.common.pytorch.model_utils.T5ForConditionalGenerationLoss) (in module modelzoo.transformers.pytorch.transformer_utils) sort() (modelzoo.common.pytorch.model_utils.checkpoint_converters.base_converter.FormatVersions method) space_timesteps() (in module modelzoo.vision.pytorch.dit.samplers.sampler_utils) SparseAttentionBuilder (class in modelzoo.transformers.utils.attention_utils) SparsityWrapperOptimizer (class in cerebras_pytorch.sparse) split_files() (in module modelzoo.transformers.data_processing.slimpajama.dedup.generate_duplicate_pairs) split_list() (in module modelzoo.transformers.data_processing.utils) split_sequences() (in module modelzoo.transformers.pytorch.t5.input.utils) split_text_and_tokenize() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors) SquadExample (class in modelzoo.transformers.data_processing.qa.qa_utils) squared_gelu() (in module modelzoo.common.pytorch.model_utils.activations) SST2DataProcessor (class in modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor) SST2Dataset (class in modelzoo.transformers.pytorch.bert.fine_tuning.classifier.input.BertClassifierDataProcessor) start_logits (modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction.RawResult attribute) start_position (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) start_time_ns (cerebras_pytorch.utils.benchmark.utils.dataloader.EpochMetrics attribute) (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics attribute) state_dict() (cerebras_pytorch.amp.GradScaler method) (cerebras_pytorch.optim.lr_scheduler.ChainedScheduler method) (cerebras_pytorch.optim.lr_scheduler.LRScheduler method) (cerebras_pytorch.optim.lr_scheduler.SequentialLR method) (cerebras_pytorch.optim.Optimizer method) (cerebras_pytorch.utils.data.DataLoader method) (cerebras_pytorch.utils.data.RestartableDataLoader method) (modelzoo.transformers.data_processing.h5_map_dataset.dataset.RestartableDataLoader method) state_names_to_sparsify() (cerebras_pytorch.optim.Adadelta method) (cerebras_pytorch.optim.Adafactor method) (cerebras_pytorch.optim.Adagrad method) (cerebras_pytorch.optim.Adamax method) (cerebras_pytorch.optim.ASGD method) (cerebras_pytorch.optim.Lamb method) (cerebras_pytorch.optim.Lion method) (cerebras_pytorch.optim.NAdam method) (cerebras_pytorch.optim.Optimizer method) (cerebras_pytorch.optim.RAdam method) (cerebras_pytorch.optim.RMSprop method) (cerebras_pytorch.optim.Rprop method) (cerebras_pytorch.optim.SGD method) StaticSparsityOptimizer (class in cerebras_pytorch.sparse) step() (cerebras_pytorch.amp.GradScaler method) (cerebras_pytorch.optim.Adadelta method) (cerebras_pytorch.optim.Adafactor method) (cerebras_pytorch.optim.Adagrad method) (cerebras_pytorch.optim.Adamax method) (cerebras_pytorch.optim.ASGD method) (cerebras_pytorch.optim.Lamb method) (cerebras_pytorch.optim.Lion method) (cerebras_pytorch.optim.lr_scheduler.LRScheduler method) (cerebras_pytorch.optim.NAdam method) (cerebras_pytorch.optim.Optimizer method) (cerebras_pytorch.optim.RAdam method) (cerebras_pytorch.optim.RMSprop method) (cerebras_pytorch.optim.Rprop method) (cerebras_pytorch.optim.SGD method) (modelzoo.vision.pytorch.dit.samplers.DDIMSampler.DDIMSampler method) (modelzoo.vision.pytorch.dit.samplers.DDPMSampler.DDPMSampler method) (modelzoo.vision.pytorch.dit.samplers.SamplerBase.SamplerBase method) step_closure() (in module cerebras_pytorch) step_if_finite() (cerebras_pytorch.amp.GradScaler method) StepLR (class in cerebras_pytorch.optim.lr_scheduler) StreamingCSLeaf (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints) StreamingCSWriter (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints) StreamingCSWriterView (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints) StreamingShardedHFReader (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints) StreamingShardedHFWriter (class in modelzoo.common.pytorch.model_utils.checkpoint_converters.streaming_checkpoints) StridedSparseAttentionBuilder (class in modelzoo.transformers.utils.attention_utils) SubsetSequentialSampler (class in modelzoo.transformers.pytorch.input_utils) SummarizationPreprocessor (class in modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors) summarize_scalar() (in module cerebras_pytorch) summarize_tensor() (in module cerebras_pytorch) SummaryReader (class in cerebras_pytorch.utils.tensorboard) SummaryWriter (class in cerebras_pytorch.utils.tensorboard) swiglu() (in module modelzoo.common.pytorch.model_utils.activations) SyntheticDataset (class in cerebras_pytorch.utils.data)

T

T5DynamicDataProcessor (class in modelzoo.transformers.pytorch.t5.input.T5DynamicDataProcessor) T5ForConditionalGeneration (class in modelzoo.transformers.pytorch.t5.t5_model) T5ForConditionalGenerationLoss (class in modelzoo.common.pytorch.model_utils.T5ForConditionalGenerationLoss) T5ForConditionalGenerationModel (class in modelzoo.transformers.pytorch.t5.model) task_id() (in module modelzoo.transformers.pytorch.input_utils) (in module modelzoo.vision.pytorch.input.utils) tensor_names() (cerebras_pytorch.utils.tensorboard.SummaryReader method) text_to_tokenized_documents() (in module modelzoo.transformers.data_processing.utils) TextFormatting (class in modelzoo.transformers.data_processing.scripts.pubmed.preprocess.TextFormatting) threshold_sample() (in module modelzoo.vision.pytorch.dit.samplers.sampler_utils) tie_weights() (modelzoo.transformers.pytorch.t5.t5_model.T5ForConditionalGeneration method) tile_image_transform() (in module modelzoo.vision.pytorch.unet.input.preprocessing_utils) tiled_decode() (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) tiled_encode() (modelzoo.vision.pytorch.dit.layers.vae.VAEModel.AutoencoderKL method) TimestepEmbeddingLayer (class in modelzoo.vision.pytorch.dit.layers.TimestepEmbeddingLayer) to_cpu() (in module modelzoo.common.pytorch.utils) to_minhash() (in module modelzoo.transformers.data_processing.slimpajama.dedup.to_hash) to_numpy() (in module cerebras_pytorch) to_tensor() (in module modelzoo.common.pytorch.utils) token_is_max_context (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) token_to_orig_map (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) tokenize() (in module modelzoo.transformers.data_processing.bert.bertsum_data_processor) (modelzoo.transformers.data_processing.tokenizers.Tokenization.BaseTokenizer method) (modelzoo.transformers.data_processing.tokenizers.Tokenization.FullTokenizer method) (modelzoo.transformers.data_processing.tokenizers.Tokenization.WordPieceTokenizer method) tokenize_dataset() (in module modelzoo.transformers.pytorch.t5.input.data_processing.preprocess_t5_dataset) tokenize_text() (in module modelzoo.transformers.pytorch.t5.input.data_processing.preprocess_t5_dataset) Tokenizer (class in modelzoo.transformers.data_processing.bert.bertsum_data_processor) tokens (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) (modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner.InputFeatures attribute)

topk() (in module cerebras_pytorch.sparse.init) torchrun() (in module modelzoo.common.pytorch.run_utils) total_steps (cerebras_pytorch.utils.benchmark.utils.dataloader.EpochMetrics attribute) (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics property) total_time (cerebras_pytorch.utils.benchmark.utils.dataloader.EpochMetrics property) (cerebras_pytorch.utils.benchmark.utils.dataloader.Metrics attribute) trace() (in module cerebras_pytorch) train_input_dataloader() (in module modelzoo.transformers.pytorch.bert.data) (in module modelzoo.transformers.pytorch.bert.fine_tuning.classifier.data) (in module modelzoo.transformers.pytorch.bert.fine_tuning.extractive_summarization.data) (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.data) (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.data) (in module modelzoo.transformers.pytorch.gpt2.data) (in module modelzoo.transformers.pytorch.layers_api_demo.data) (in module modelzoo.transformers.pytorch.t5.data) (in module modelzoo.transformers.pytorch.transformer.data) (in module modelzoo.vision.pytorch.dit.data) (in module modelzoo.vision.pytorch.unet.data) training_data_generator() (in module modelzoo.transformers.pytorch.gpt2.input.scripts.data_processor_utils) transform_image_and_mask() (modelzoo.vision.pytorch.unet.input.SkmDataProcessor.SkmDataProcessor method) Transformer (class in modelzoo.common.pytorch.layers) TransformerBaseModel (class in modelzoo.transformers.pytorch.layers_api_demo.model) TransformerDecoder (class in modelzoo.common.pytorch.layers) TransformerDecoderLayer (class in modelzoo.common.pytorch.layers) TransformerDynamicDataProcessor (class in modelzoo.transformers.pytorch.transformer.input.TransformerDynamicDataProcessor) TransformerEncoder (class in modelzoo.common.pytorch.layers) TransformerEncoderLayer (class in modelzoo.common.pytorch.layers) TransformerModel (class in modelzoo.transformers.pytorch.layers_api_demo.cb_transformer) (class in modelzoo.transformers.pytorch.layers_api_demo.pytorch_transformer) transpose_key_if_2D() (in module modelzoo.common.pytorch.model_utils.checkpoint_converters.gpt2_hf_cs) trunc_normal_() (in module modelzoo.common.pytorch.model_utils.weight_initializers)

U

UNet (class in modelzoo.vision.pytorch.unet.modeling_unet) UNetBlock (class in modelzoo.vision.pytorch.unet.layers.UNetBlock) UNetDataProcessor (class in modelzoo.vision.pytorch.unet.input.UNetDataProcessor) UNetMidBlock2D (class in modelzoo.vision.pytorch.dit.layers.vae.UNetMidBlock2D) UNetModel (class in modelzoo.vision.pytorch.unet.model) unique_id (modelzoo.transformers.data_processing.qa.qa_utils.InputFeatures attribute) (modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction.RawResult attribute) unscale_() (cerebras_pytorch.amp.GradScaler method) update() (cerebras_pytorch.amp.GradScaler method) (cerebras_pytorch.metrics.AccuracyMetric method) (cerebras_pytorch.metrics.DiceCoefficientMetric method) (cerebras_pytorch.metrics.FBetaScoreMetric method) (cerebras_pytorch.metrics.MeanIOUMetric method) (cerebras_pytorch.metrics.Metric method) (cerebras_pytorch.metrics.PerplexityMetric method) (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) update_debug_args_with_mem_limits() (in module modelzoo.common.input.utils)

update_defaults() (in module modelzoo.common.run_utils.cli_parser) update_groups() (cerebras_pytorch.optim.lr_scheduler.LRScheduler method) update_last_lr() (cerebras_pytorch.optim.lr_scheduler.LRScheduler method) update_mask() (cerebras_pytorch.sparse.dynamic.DynamicSparsityOptimizer method) update_params() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) update_params_from_args() (in module modelzoo.common.run_utils.cli_parser) (in module modelzoo.vision.pytorch.unet.input.scripts.create_hdf5_files) update_params_from_file() (in module modelzoo.common.run_utils.cli_parser) update_parser() (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner) update_runconfig_debug_args_path() (in module modelzoo.vision.pytorch.utils.run_utils) update_scale() (cerebras_pytorch.amp.GradScaler method) update_sideband_mode_arg() (in module modelzoo.common.pytorch.run_utils) UpDecoderBlock2D (class in modelzoo.vision.pytorch.dit.layers.vae.UpDecoderBlock2D) Upsample2D (class in modelzoo.vision.pytorch.dit.layers.vae.ResNetBlock2D) use_bfloat16() (in module cerebras_pytorch.amp) use_cs() (in module cerebras_pytorch) utf8len() (in module modelzoo.transformers.data_processing.slimpajama.utils.utils)

V

validate() (modelzoo.common.pytorch.utils.RunConfigParamsValidator method) validate_streaming_and_micro_batch_size() (in module modelzoo.common.pytorch.input_utils) validate_tokens() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) values() (modelzoo.transformers.pytorch.bert.input.utils.Vocab method) variance_scaling_() (in module modelzoo.common.pytorch.model_utils.weight_initializers) VerificationArgs (class in modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) verify_saved_hdf5_files() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) verify_saved_hdf5_files_mp() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) VisionSubset (class in modelzoo.vision.pytorch.input.classification.dataset_factory)

visit_state() (cerebras_pytorch.optim.Optimizer method) (cerebras_pytorch.sparse.base.BaseSparsityOptimizer method) (cerebras_pytorch.sparse.dynamic.DynamicSparsityOptimizer method) (cerebras_pytorch.sparse.SparsityWrapperOptimizer method) (cerebras_pytorch.sparse.utils.BaseHyperParameter method) visit_structure() (in module modelzoo.common.pytorch.utils) visualize_dataset() (in module modelzoo.vision.pytorch.unet.input.UNetDataProcessor) ViTEmbeddingLayer (class in modelzoo.common.pytorch.layers) Vocab (class in modelzoo.transformers.pytorch.bert.input.utils)

W

weight (modelzoo.common.pytorch.layers.BiaslessLayerNorm attribute) whitespace_tokenize() (in module modelzoo.transformers.data_processing.utils) wikitext_detokenizer() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.utils) WordPieceTokenizer (class in modelzoo.transformers.data_processing.tokenizers.Tokenization) write_csv_files() (in module modelzoo.transformers.data_processing.qa.write_csv_qa) (in module modelzoo.transformers.pytorch.bert.fine_tuning.token_classifier.input.write_csv_ner) write_docs() (in module modelzoo.transformers.data_processing.slimpajama.preprocessing.shuffle_holdout) write_hdf5_file() (in module modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.convert_dataset_to_HDF5) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method)

write_hdf5_files() (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_base_preprocessor.HDF5BasePreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_curation_corpus_preprocessor.CurationCorpusPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.LMDataPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_dataset_preprocessors.SummarizationPreprocessor method) (modelzoo.transformers.data_processing.scripts.hdf5_preprocessing.hdf5_nlg_preprocessor.NLGPreprocessor method) write_label_map_files() (in module modelzoo.transformers.data_processing.bert.ner_data_processor) write_lmd_dataset() (in module modelzoo.transformers.data_processing.slimpajama.utils.utils) write_predictions() (in module modelzoo.transformers.pytorch.bert.fine_tuning.qa.run_prediction) write_samples() (in module modelzoo.transformers.data_processing.slimpajama.split_dataset)

Z

zero_grad() (cerebras_pytorch.sparse.base.BaseSparsityOptimizer method) (cerebras_pytorch.sparse.RigLSparsityOptimizer method)

zeros() (in module cerebras_pytorch) zeros_like() (in module cerebras_pytorch)