Metadata-Version: 2.5
Name: canvit-pytorch
Version: 0.2.0
Summary: CanViT (Canvas Vision Transformer): model, pretraining, task specialization and evaluation
Project-URL: Repository, https://github.com/m2b3/CanViT
Project-URL: Homepage, https://m2b3.github.io/CanViT/
Project-URL: Paper, https://arxiv.org/abs/2603.22570
Project-URL: Checkpoints, https://huggingface.co/canvit
Author-email: Yohaï-Eliel Berreby <me@yberreby.com>, Sabrina Du <sabrina.du@mail.mcgill.ca>
License-Expression: MIT
Keywords: active-vision,canvit,foundation-model,pytorch,vision-transformer
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.12
Requires-Dist: huggingface-hub>=1.3.2
Requires-Dist: numpy<2.4.0,>=2.2.0
Requires-Dist: safetensors>=0.7.0
Requires-Dist: torch>=2.14.0; sys_platform == 'darwin'
Requires-Dist: torch>=2.9.0
Requires-Dist: torchvision>=0.22.0
Provides-Extra: demo
Requires-Dist: matplotlib>=3.10.8; extra == 'demo'
Requires-Dist: pillow>=10.0.0; extra == 'demo'
Requires-Dist: tyro>=1.0.3; extra == 'demo'
Provides-Extra: evaluate
Requires-Dist: pandas>=2.0.0; extra == 'evaluate'
Requires-Dist: pillow>=10.0.0; extra == 'evaluate'
Requires-Dist: pyarrow>=20.0.0; extra == 'evaluate'
Requires-Dist: tqdm>=4.60.0; extra == 'evaluate'
Requires-Dist: transformers>=4.50.0; extra == 'evaluate'
Requires-Dist: tyro>=1.0.0; extra == 'evaluate'
Provides-Extra: live
Requires-Dist: onnx>=1.23.0; extra == 'live'
Requires-Dist: onnxruntime>=1.30.0; extra == 'live'
Requires-Dist: onnxscript>=0.7.2; extra == 'live'
Requires-Dist: pillow>=10.0.0; extra == 'live'
Requires-Dist: playwright>=1.63.0; extra == 'live'
Requires-Dist: tyro>=1.0.0; extra == 'live'
Provides-Extra: pretrain
Requires-Dist: comet-ml; extra == 'pretrain'
Requires-Dist: matplotlib; extra == 'pretrain'
Requires-Dist: pyarrow; extra == 'pretrain'
Requires-Dist: torch>=2.10; extra == 'pretrain'
Requires-Dist: transformers>=4.50.0; extra == 'pretrain'
Requires-Dist: tyro; extra == 'pretrain'
Requires-Dist: xxhash; extra == 'pretrain'
Provides-Extra: specialize
Requires-Dist: comet-ml>=3.30.0; extra == 'specialize'
Requires-Dist: dinov3; extra == 'specialize'
Requires-Dist: matplotlib>=3.8.0; extra == 'specialize'
Requires-Dist: pillow>=10.0.0; extra == 'specialize'
Requires-Dist: tqdm>=4.60.0; extra == 'specialize'
Requires-Dist: transformers>=4.50.0; extra == 'specialize'
Requires-Dist: tyro>=1.0.0; extra == 'specialize'
Provides-Extra: viz
Requires-Dist: pillow>=10.0.0; extra == 'viz'
Requires-Dist: tyro>=1.0.0; extra == 'viz'
Description-Content-Type: text/markdown

# canvit-pytorch

PyTorch implementation of CanViT, the Canvas Vision Transformer, from
[CanViT: Toward Active-Vision Foundation Models](https://arxiv.org/abs/2603.22570) (NeurIPS 2026).

```bash
pip install canvit-pytorch
```

Checkpoints: <https://huggingface.co/canvit>. Documentation and examples: <https://github.com/m2b3/CanViT>.

MIT license.
