MCPcopy Create free account

hub / github.com/ByteDance-Seed/AHN / types & classes

Types & classes48 in github.com/ByteDance-Seed/AHN

↓ 8 callersClassQwen3RMSNorm
src/ahn/transformer/qwen3/modeling_qwen3.py:63
↓ 6 callersClassQwen2RMSNorm
src/ahn/transformer/qwen2/modeling_qwen2.py:211
↓ 4 callersClassGroupLinear
src/ahn/utils.py:29
↓ 4 callersClassQwen2Model
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`Qwen2DecoderLayer`] Args: config: Qwen2Config
src/ahn/transformer/qwen2/modeling_qwen2.py:440
↓ 4 callersClassQwen3Model
Transformer decoder consisting of *config.num_hidden_layers* layers. Each layer is a [`Qwen3DecoderLayer`] Args: config: Qwen3Config
src/ahn/transformer/qwen3/modeling_qwen3.py:468
↓ 2 callersClassAHNRouter
src/ahn/utils.py:183
↓ 2 callersClassBaseAHN
src/ahn/utils.py:82
↓ 2 callersClassQwen2MLP
src/ahn/transformer/qwen2/modeling_qwen2.py:47
↓ 2 callersClassQwen2RotaryEmbedding
src/ahn/transformer/qwen2/modeling_qwen2.py:288
↓ 2 callersClassQwen3MLP
src/ahn/transformer/qwen3/modeling_qwen3.py:83
↓ 2 callersClassQwen3RotaryEmbedding
src/ahn/transformer/qwen3/modeling_qwen3.py:316
↓ 1 callersClassCustomizedCausalLMOutputWithPast
src/ahn/transformer/qwen2_ahn/qwen2_ahn.py:57
↓ 1 callersClassCustomizedCausalLMOutputWithPast
src/ahn/transformer/qwen3_ahn/qwen3_ahn.py:59
↓ 1 callersClassCustomizedModelOutputWithPast
src/ahn/transformer/qwen2_ahn/qwen2_ahn.py:46
↓ 1 callersClassCustomizedModelOutputWithPast
src/ahn/transformer/qwen3_ahn/qwen3_ahn.py:48
↓ 1 callersClassQwen2Attention
Multi-headed attention from 'Attention Is All You Need' paper
src/ahn/transformer/qwen2/modeling_qwen2.py:135
↓ 1 callersClassQwen2DecoderLayer
src/ahn/transformer/qwen2/modeling_qwen2.py:231
↓ 1 callersClassQwen2MemFlexAttn
src/ahn/transformer/qwen2_ahn/qwen2_ahn.py:112
↓ 1 callersClassQwen2Model
src/ahn/transformer/qwen2_ahn/qwen2_ahn.py:382
↓ 1 callersClassQwen3Attention
Multi-headed attention from 'Attention Is All You Need' paper
src/ahn/transformer/qwen3/modeling_qwen3.py:171
↓ 1 callersClassQwen3DecoderLayer
src/ahn/transformer/qwen3/modeling_qwen3.py:257
↓ 1 callersClassQwen3MemFlexAttn
src/ahn/transformer/qwen3_ahn/qwen3_ahn.py:114
↓ 1 callersClassQwen3Model
src/ahn/transformer/qwen3_ahn/qwen3_ahn.py:385
ClassDeltaNet
r""" The layer implementaion for [Parallelizing Linear Transformers with the Delta Rule over Sequence Length](https://arxiv.org/abs/2406.06484).
src/ahn/rnn/delta_net.py:41
ClassGatedDeltaNet
The layer implementaion for [Gated Delta Networks: Improving Mamba2 with Delta Rule](https://arxiv.org/abs/2412.06464). # noqa Similar to M
src/ahn/rnn/gated_deltanet.py:46
ClassKwargsForCausalLM
src/ahn/transformer/qwen3/modeling_qwen3.py:826
ClassKwargsForCausalLM
src/ahn/transformer/qwen2/modeling_qwen2.py:732
ClassMamba2
Compute ∆, A, B, C, and D the state space parameters and compute the `contextualized_states`. A, D are input independent (see Mamba paper [1]
src/ahn/rnn/mamba2.py:112
ClassQwen2Config
r""" This is the configuration class to store the configuration of a [`Qwen2Model`]. It is used to instantiate a Qwen2 model according to the
src/ahn/transformer/qwen2/configuration_qwen2.py:25
ClassQwen2Config
r""" Args: _layer_implementation (`str`, *optional*, defaults to 'Qwen2DecoderLayer'): Decoder layer implementation. _
src/ahn/transformer/qwen2_ahn/qwen2_ahn.py:66
ClassQwen2ForCausalLM
src/ahn/transformer/qwen2/modeling_qwen2.py:735
ClassQwen2ForCausalLM
src/ahn/transformer/qwen2_ahn/qwen2_ahn.py:717
ClassQwen2ForQuestionAnswering
src/ahn/transformer/qwen2/modeling_qwen2.py:1044
ClassQwen2ForSequenceClassification
src/ahn/transformer/qwen2/modeling_qwen2.py:869
ClassQwen2ForTokenClassification
src/ahn/transformer/qwen2/modeling_qwen2.py:962
ClassQwen2MemDecoderLayer
src/ahn/transformer/qwen2_ahn/qwen2_ahn.py:1048
ClassQwen2PreTrainedModel
src/ahn/transformer/qwen2/modeling_qwen2.py:343
ClassQwen2Tokenizer
Construct a Qwen2 tokenizer. Based on byte-level Byte-Pair-Encoding. Same with GPT2Tokenizer, this tokenizer has been trained to treat space
src/ahn/transformer/qwen2/tokenization_qwen2.py:83
ClassQwen2TokenizerFast
Construct a "fast" Qwen2 tokenizer (backed by HuggingFace's *tokenizers* library). Based on byte-level Byte-Pair-Encoding. Same with GPT
src/ahn/transformer/qwen2/tokenization_qwen2_fast.py:37
ClassQwen3Config
r""" This is the configuration class to store the configuration of a [`Qwen3Model`]. It is used to instantiate a Qwen3 model according to the
src/ahn/transformer/qwen3/configuration_qwen3.py:25
ClassQwen3Config
r""" Args: _layer_implementation (`str`, *optional*, defaults to 'Qwen3DecoderLayer'): Decoder layer implementation. _
src/ahn/transformer/qwen3_ahn/qwen3_ahn.py:68
ClassQwen3ForCausalLM
src/ahn/transformer/qwen3/modeling_qwen3.py:829
ClassQwen3ForCausalLM
src/ahn/transformer/qwen3_ahn/qwen3_ahn.py:712
ClassQwen3ForQuestionAnswering
src/ahn/transformer/qwen3/modeling_qwen3.py:1138
ClassQwen3ForSequenceClassification
src/ahn/transformer/qwen3/modeling_qwen3.py:963
ClassQwen3ForTokenClassification
src/ahn/transformer/qwen3/modeling_qwen3.py:1056
ClassQwen3MemDecoderLayer
src/ahn/transformer/qwen3_ahn/qwen3_ahn.py:1033
ClassQwen3PreTrainedModel
src/ahn/transformer/qwen3/modeling_qwen3.py:371