Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | __init__.py | 225 | logplain |
-rw-r--r-- | attention.py | 3999 | logplain |
-rw-r--r-- | attention_layers.py | 412 | logplain |
-rw-r--r-- | mlp.py | 1271 | logplain |
-rw-r--r-- | norm.py | 1276 | logplain |
d--------- | nystromer | 119 | logplain |
-rw-r--r-- | positional_encoding.py | 3102 | logplain |
-rw-r--r-- | rotary_embedding.py | 1194 | logplain |
-rw-r--r-- | transformer.py | 8198 | logplain |
-rw-r--r-- | vit.py | 0 | logplain |