Skip to main content

Gated Slot Attention

Join our Discord Subscribe on YouTube Connect on LinkedIn Follow on X.com

Usage

import torch
from gated_slot_attention.model import GSATransformer

model = GSATransformer(
    dim=512,
    heads=8,
    m=64,
    tau=0.1,
    depth=1,
    vocab_size=10000,
    max_seq_len=1024,
)

x = torch.randint(0, 10000, (1, 1024))
out = model(x)
print(out.shape)

License

MIT

Metadata

Release files for gated-slot-attention 0.0.1

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for gated-slot-attention 0.0.1
File Size Uploaded
gated_slot_attention-0.0.1.tar.gz 5.1 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for gated-slot-attention 0.0.1
File Interpreter ABI Platform
gated_slot_attention-0.0.1-py3-none-any.whl Python 3 none any Details

Total release size: 10.7 kB

Release files / gated_slot_attention-0.0.1.tar.gz

Download URL gated_slot_attention-0.0.1.tar.gz
Size 5.1 kB
Tags Source
SHA-256 checksum
How to use checksums
be2758fd001a81b1ff1f3ababc62e09231d826fea387d43bbd3a7eecc9f206b6
BLAKE2b-256 checksum
How to use checksums
6d2204728f8c974bf56e6c56dc7cf58368a6a7613f2e25e3d2605e1b57ca816e
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via poetry/1.8.3 CPython/3.12.3 Darwin/23.3.0

Release files / gated_slot_attention-0.0.1-py3-none-any.whl

Download URL gated_slot_attention-0.0.1-py3-none-any.whl
Size 5.6 kB
Tags Python 3
SHA-256 checksum
How to use checksums
5afce2322c21ceba30ccf73ee08c144fb7e3336f475881f098378a3bd37324c8
BLAKE2b-256 checksum
How to use checksums
b8c257db15f9525977620c41875673aeb7a302638f5b3844ac39185f39023e30
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via poetry/1.8.3 CPython/3.12.3 Darwin/23.3.0

Release history Release notifications | RSS feed

This release

0.0.1 This release

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page