Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | __init__.py | 222 | logplain |
-rw-r--r-- | attention.py | 2097 | logplain |
-rw-r--r-- | mlp.py | 1271 | logplain |
-rw-r--r-- | norm.py | 947 | logplain |
-rw-r--r-- | positional_encoding.py | 3102 | logplain |
-rw-r--r-- | rotary_embedding.py | 1194 | logplain |
-rw-r--r-- | transformer.py | 7731 | logplain |