torch-einops-utils
Some utility functions to help myself (and perhaps others) go faster with ML/AI work
maybe_return
return just the primary output by default, extras only on request
from torch_einops_utils import maybe_return
@maybe_return('memories')
def forward(self, ids, return_memories = False):
memories = build_memories(ids) if return_memories else None
return backbone(ids), memories
model(ids) # logits
model(ids, return_memories = True) # Output(logits, memories)
logits, memories = model(ids, return_memories = True) # unpacks
any number of extras, individually or all at once
@maybe_return('memories', 'hiddens')
def forward(self, ids, return_memories = False, return_hiddens = False):
...
return logits, memories, hiddens
model(ids, return_hiddens = True) # Output(logits, hiddens)
model(ids, return_all = True) # Output(logits, memories, hiddens)
primary may be a tuple of names, or a callable receiving the bound call arguments, for when the shape of the primary depends on the mode
def primary_fields(kwargs):
if not kwargs.get('return_loss', False):
return ('logits', 'memory') if kwargs.get('return_memory', False) else 'logits'
return ('loss', 'loss_mask') if not kwargs.get('reduce_loss', False) else 'loss'
@maybe_return('pooled_repr', primary = primary_fields)
def forward(self, ids, return_loss = False, return_memory = False, reduce_loss = False, return_pooled_repr = False):
...
if return_loss:
return loss, loss_mask, pooled_repr
return logits, memory, pooled_repr
model(ids) # logits
logits, memory = model(ids, return_memory = True)
loss, loss_mask = model(ids, return_loss = True)
loss = model(ids, return_loss = True, reduce_loss = True)
loss, pooled = model(ids, return_loss = True, reduce_loss = True, return_pooled_repr = True)
1 maintainer
Metadata
1 maintainer
Release files for torch-einops-utils 0.1.37
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | Provenance | |
|---|---|---|---|---|
| torch_einops_utils-0.1.37.tar.gz | 28.5 kB | Without provenance | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| torch_einops_utils-0.1.37-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 47.0 kB
Release files / torch_einops_utils-0.1.37.tar.gz
| Download URL | torch_einops_utils-0.1.37.tar.gz |
|---|---|
| Size | 28.5 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
41bee987f326b039c11468eac3acc6c58ecb37255707c61bfd89a14728c40dfe
|
|
BLAKE2b-256 checksum How to use checksums |
2bc17771bc8ccda6ec5b389759b3149a05304bc5e0fb2f9f91ce8f20bca6a942
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
uv/0.8.17
|
Release files / torch_einops_utils-0.1.37-py3-none-any.whl
| Download URL | torch_einops_utils-0.1.37-py3-none-any.whl |
|---|---|
| Size | 18.5 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
5ad455075d481cd1adb142c41ebfe3e2942e8c6289996cbf79a0aa40708e6d8d
|
|
BLAKE2b-256 checksum How to use checksums |
f6f104cfe12d5b35dfb900b411a3e2bcea027352c82befea2dca7f8e7f61b116
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
uv/0.8.17
|
Security for torch-einops-utils 0.1.37
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.
