JAX ports of popular models(audio, vision, llm, diffusion, etc)
kelechic
tensorkelechi
AI & ML interests
vision
Recent Activity
updated
a model
about 13 hours ago
tensorkelechi/dinov2-small-jax
upvoted
a
paper
1 day ago
MusicInfuser: Making Video Diffusion Listen and Dance
Organizations
Collections
8
spaces
10
models
11

tensorkelechi/dinov2-small-jax
Updated

tensorkelechi/whisper_base_jax
Updated

tensorkelechi/microdit_b
Updated

tensorkelechi/microdit_cifar
Updated

tensorkelechi/signature_mobilenet
Updated

tensorkelechi/canary_mini
Text-to-Audio
•
Updated

tensorkelechi/sky_diffuse
Unconditional Image Generation
•
Updated
•
98

tensorkelechi/katara-mini
Updated

tensorkelechi/leaf_disease_mobilenet
Updated
•
6

tensorkelechi/HAR_mobilenet
Image Classification
•
Updated
•
24
•
1
datasets
13
tensorkelechi/latent_flowers102
Viewer
•
Updated
•
1.02k
•
83
•
1
tensorkelechi/tinywebvid
Viewer
•
Updated
•
8.19k
•
129
tensorkelechi/tiny_webvid_latents
Viewer
•
Updated
•
64
•
89
tensorkelechi/tiny_webvid
Viewer
•
Updated
•
1.02k
•
240
tensorkelechi/virat2
Viewer
•
Updated
•
12.8k
•
77
tensorkelechi/repo_id
Viewer
•
Updated
•
12.8k
•
69
tensorkelechi/laion_moondream_caption_10k
Updated
•
38
tensorkelechi/fma_medium
Updated
•
42
tensorkelechi/freesound_mini
Viewer
•
Updated
•
100
•
69
tensorkelechi/freesound_audio_sm
Viewer
•
Updated
•
7.45k
•
93