Packages

TorchVision ops for ExTorch — NMS, ROI Align, deformable convolution, and image I/O on the BEAM. Builds libtorchvision from source, no C++ code in this package.

Current section

Files

Jump to
extorch_vision CHANGELOG.md
Raw

CHANGELOG.md

# Changelog
## 0.1.0 (2026-04-11)
Initial release.
### Features
- **Detection ops**: `nms/3`, `roi_align/7`, `roi_pool/5`, `ps_roi_align/6`, `ps_roi_pool/5`, `deform_conv2d/14`
- **Image I/O**: `decode_jpeg/3`, `encode_jpeg/2`, `decode_png/3`, `encode_png/2`, `decode_webp/2`, `decode_gif/1`, `decode_image/3`
- **GPU image ops**: `decode_jpegs_cuda/3`, `encode_jpegs_cuda/2` (NVJPEG)
- **Auto-build**: `libtorchvision.so` built from source via CMake at compile time, no manual steps
- **ExTorch.Export integration**: All ops registered with `OpRegistry` for transparent use in exported model inference
- **Zero C++/Rust code**: Everything dispatched through `ExTorch.Native.dispatch_op/3`