ViT JAX
ViT JAX is Google's image to text model. A Vision Transformer model implemented in JAX for high-performance image classification with hardware-accelerated training and inference.
google-vit-jax |
| Image to Text |
| Active |
| Google |
| Image |
| Text |
Capabilities
Input1/5
Text·
Image✓
Audio·
Video·
PDF·
Output1/5
Text✓
Image·
Audio·
Video·
Embedding·
Capabilities0/13
Reasoning·
Adaptive Reasoning·
Function Calling·
Parallel Function Calling·
Structured Outputs·
Native JSON Schema·
Web Search·
URL Context·
Computer Use·
Code Execution·
File Search·
Prompt Caching·
Assistant Prefill·
Versions
| Version | Released | Context | Input / 1M | Output / 1M | Status |
|---|---|---|---|---|---|
| ViT JAX | — | — | — | — | Current |
| ViT Image Classification | — | — | — | — | Available |
| ViT Image Classification Proprietary | — | — | — | — | Available |