hub / github.com/idiap/fast-transformers

github.com/idiap/fast-transformers @v0.4.0 sqlite

repository ↗ · DeepWiki ↗ · release v0.4.0 ↗

766 symbols 2,750 edges 122 files 176 documented · 23%

Core symbols most depended-on inside this repo

backward

called by 59

fast_transformers/local_product/__init__.py

get

called by 59

fast_transformers/attention_registry/spec.py

aggregate

called by 36

fast_transformers/aggregate/__init__.py

broadcast

called by 31

fast_transformers/aggregate/__init__.py

dropout

called by 24

fast_transformers/builders/transformer_builders.py

cluster

called by 22

fast_transformers/clustering/hamming/__init__.py

keys

called by 21

fast_transformers/attention_registry/registry.py

compute_hashes

called by 17

fast_transformers/hashing/__init__.py

Shape

Method 516

Class 143

Function 57

Route 50

Languages

Python100%

Modules by API surface

fast_transformers/builders/transformer_builders.py61 symbols

fast_transformers/weight_mapper.py32 symbols

tests/local_product/test_local_product_cuda.py22 symbols

fast_transformers/attention_registry/spec.py22 symbols

fast_transformers/masking.py17 symbols

fast_transformers/events/filters.py17 symbols

fast_transformers/feature_maps/fourier_features.py15 symbols

fast_transformers/builders/attention_builders.py15 symbols

tests/clustering/hamming/test_python_api_gpu.py14 symbols

tests/causal_product/test_causal_product_gpu.py13 symbols

fast_transformers/attention/improved_clustered_attention.py13 symbols

tests/sparse_product/test_clustered_sparse_weighted_average_gpu.py12 symbols

For agents

$ claude mcp add fast-transformers \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact