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 Provenance Actions
gated_slot_attention-0.0.1.tar.gz 5.1 kB Without provenance Details for gated_slot_attention-0.0.1.tar.gz

Built distribution (wheel)

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

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

Security for gated-slot-attention 0.0.1

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.

Release history Release notifications | RSS feed

This release

0.0.1 This release

2 release files No provenance

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