MCPcopy Create free account

hub / github.com/arcee-ai/DistillKit / types & classes

Types & classes36 in github.com/arcee-ai/DistillKit

⨍ Functions 146◇ Types & classes 36↳ Endpoints 2
↓ 11 callersClassDistributionQuantizationConfig
distillkit/compression/config.py:65
↓ 4 callersClassQuantizationBin
distillkit/compression/config.py:43
↓ 3 callersClassLogprobCompressor
distillkit/compression/compressor.py:23
↓ 2 callersClassLegacyLogitCompressionConfig
Configuration for legacy polynomial logit compression. Args: k (int): Total number of logits per token exact_k (int): Number of e
distillkit/compression/config.py:150
↓ 2 callersClassLossFunctionConfig
distillkit/configuration.py:26
↓ 2 callersClassSparseSignal
distillkit/signals.py:25
↓ 1 callersClassCompressionEvalResult
test_compression.py:45
↓ 1 callersClassDenseSignal
distillkit/signals.py:20
↓ 1 callersClassDistillationTrainer
distillkit/trainer.py:24
↓ 1 callersClassHiddenStateMapping
distillkit/hsd_mapping.py:6
↓ 1 callersClassOfflineSignalSource
distillkit/signals.py:44
↓ 1 callersClassOnlineSignalSource
distillkit/signals.py:86
↓ 1 callersClassStreamingParquetWriter
distillkit/sample_common.py:158
ClassCrossEntropyLoss
distillkit/lossfuncs/cross_entropy.py:12
ClassDatasetConfiguration
distillkit/configuration.py:84
ClassDistillationRunConfig
distillkit/configuration.py:139
ClassErrorStats
test_compression.py:21
ClassHfRepoDataset
distillkit/configuration.py:53
ClassHiddenStateCosineLoss
distillkit/lossfuncs/hidden_state.py:65
ClassHiddenStateMSELoss
distillkit/lossfuncs/hidden_state.py:92
ClassHingeLoss
distillkit/lossfuncs/hingeloss.py:64
ClassJSDLoss
distillkit/lossfuncs/jsd.py:281
ClassKLDLoss
distillkit/lossfuncs/kl.py:175
ClassLocalDataset
distillkit/configuration.py:71
ClassLogisticRankingLoss
distillkit/lossfuncs/logistic_ranking.py:96
ClassLogitCompressor
Compresses and decompresses logits using polynomial approximation.
distillkit/compression/legacy.py:11
ClassLossFunction
distillkit/configuration.py:15
ClassLossFunctionBase
distillkit/lossfuncs/common.py:185
ClassMissingProbabilityHandling
distillkit/missing_probability.py:4
ClassSignalSource
distillkit/signals.py:34
ClassSpecialTerm
distillkit/compression/config.py:7
ClassTVDLoss
distillkit/lossfuncs/tvd.py:188
ClassTeacherDatasetConfig
distillkit/configuration.py:127
ClassTeacherModelConfig
distillkit/configuration.py:118
ClassTeacherSignalBase
distillkit/signals.py:13
ClassTermDtype
distillkit/compression/config.py:12