JaxNN/resnet18.fb_swsl_ig1b_ft_in1k
Image Classification • Updated • 85
AI & ML interests
None defined yet.
Recent 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 67
JaxNN/resnet14t.c3_in1k
Image Classification • Updated • 81
JaxNN/resnet152d.ra2_in1k
Image Classification • Updated • 95
JaxNN/resnet152d.gluon_in1k
Image Classification • Updated • 29
JaxNN/resnet152.tv_in1k
Image Classification • Updated • 25
JaxNN/resnet101d.gluon_in1k
Image Classification • Updated • 27
JaxNN/resnet101.tv_in1k
Image Classification • Updated • 25
JaxNN/resnet50d.gluon_in1k
Image Classification • Updated • 31
JaxNN/resnet34.ra4_e3600_r224_in1k
Image Classification • Updated • 53
JaxNN/resnet50_gn.a1h_in1k
Image Classification • Updated • 46
datasets 0
None public yet