Mode | Name | Size | |
---|---|---|---|
-rw-r--r-- | __init__.py | 225 | logplain |
-rw-r--r-- | attention.py | 2331 | 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 | 8198 | logplain |