Metadata-Version: 2.4
Name: sensa
Version: 0.1.0
Summary: ML Training Library.
Project-URL: Documentation, https://github.com/Tensor46/sensa#readme
Project-URL: Issues, https://github.com/Tensor46/sensa/issues
Project-URL: Source, https://github.com/Tensor46/sensa
Project-URL: Changelog, https://github.com/Tensor46/sensa/blob/main/CHANGELOG.md
Author-email: Tensor46 <vikas11187@gmail.com>
License-Expression: MIT
License-File: LICENSE
Keywords: Lightning,MAE,ML-Training,VIT
Classifier: Development Status :: 5 - Production/Stable
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python
Classifier: Programming Language :: Python :: 3.10
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: Implementation :: CPython
Classifier: Programming Language :: Python :: Implementation :: PyPy
Requires-Python: <3.13,>=3.10
Requires-Dist: einops; python_version < '3.13.0'
Requires-Dist: lightning; python_version < '3.13.0'
Requires-Dist: numba>=0.50.0; python_version < '3.13.0'
Requires-Dist: numpy>=1.23.0; python_version < '3.13.0'
Requires-Dist: pillow>=10.0.0; python_version < '3.13.0'
Requires-Dist: pydantic>=2.0.0; python_version < '3.13.0'
Requires-Dist: pyyaml; python_version < '3.13.0'
Requires-Dist: torch; python_version < '3.13.0'
Requires-Dist: torchvision; python_version < '3.13.0'
Description-Content-Type: text/markdown

# Sensa
[![Test & Build](https://github.com/Tensor46/sensa/actions/workflows/python-package.yml/badge.svg?branch=main)](https://github.com/Tensor46/sensa/actions/workflows/python-package.yml)

Sensa (from the Latin sēnsus, meaning “perception”): A PyTorch Lightning based library providing Masked Autoencoder (MAE) pretraining and classification training using customizable Vision Transformer (ViT) backbones.

### Features
* Masked Autoencoder Pretraining with random token masking and reconstruction.
* Classification Training with flexible pooling strategies (avg, full, half, token).

### License
MIT License
