- Implementation of the basic Vision Transformer using
jaxanddm-haiku - Model and optimizer serialization/deserialization with pickle
- Uses tensor manipulation operations using
einops - Re-implementation of Attention Rollout with percentile based filtering from
https://github.com/jacobgil/vit-explain(WIP) - Inference script with attention rollout (WIP)
generated from dogeplusplus/python-template
-
Notifications
You must be signed in to change notification settings - Fork 0
Vision Transformer implemented with JAX and dm-haiku
License
dogeplusplus/haiku-transformer
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
About
Vision Transformer implemented with JAX and dm-haiku
Topics
Resources
License
Stars
Watchers
Forks
Releases
No releases published
Packages 0
No packages published