JaxNN/resnet14t.c3_in1k
Image Classification • Updated • 2
AI & ML interests
None defined yet.
Recent Activity
View all activity
Organization Card
JaxNN: Foundation Models in JAX/Flax
Jax Neural Networks is an open-source library for foundation models in JAX and Flax. It provides a unified framework for loading, creating, and using pretrained models (e.g., ResNet, ViT).
models 92
JaxNN/vit_large_patch16_224.augreg_in21k
Image Classification • Updated
JaxNN/vit_base_patch16_224.augreg2_in21k_ft_in1k
Image Classification • Updated
JaxNN/vit_large_patch16_224.mae
Image Classification • Updated
JaxNN/vit_small_patch16_224.dino
Image Classification • Updated • 1
JaxNN/vit_base_patch32_224.augreg_in1k
Image Classification • Updated
JaxNN/vit_base_patch16_224.sam_in1k
Image Classification • Updated
JaxNN/vit_base_patch32_224.augreg_in21k_ft_in1k
Image Classification • Updated
JaxNN/vit_base_patch16_224.orig_in21k
Image Classification • Updated
JaxNN/vit_large_patch16_224.orig_in21k
Image Classification • Updated
datasets 0
None public yet