JAX ports of popular models(audio, vision, llm, diffusion, etc)
kelechic
tensorkelechi
AI & ML interests
vision
Recent Activity
updated
a collection
3 days ago
SAE
updated
a collection
16 days ago
SAE
updated
a collection
16 days ago
SAE
Organizations
models
11
tensorkelechi/dinov2-small-jax
Updated
tensorkelechi/whisper_base_jax
Updated
tensorkelechi/microdit_b
Updated
tensorkelechi/microdit_cifar
Updated
tensorkelechi/signature_mobilenet
0.0B
•
Updated
tensorkelechi/canary_mini
Text-to-Audio
•
Updated
tensorkelechi/sky_diffuse
Unconditional Image Generation
•
Updated
•
50
tensorkelechi/katara-mini
Updated
tensorkelechi/leaf_disease_mobilenet
0.0B
•
Updated
•
8
tensorkelechi/HAR_mobilenet
Image Classification
•
0.0B
•
Updated
•
31
•
1
datasets
13
tensorkelechi/latent_flowers102
Viewer
•
Updated
•
1.02k
•
29
•
1
tensorkelechi/tinywebvid
Viewer
•
Updated
•
8.19k
•
640
tensorkelechi/tiny_webvid_latents
Viewer
•
Updated
•
64
•
30
tensorkelechi/tiny_webvid
Viewer
•
Updated
•
1.02k
•
141
tensorkelechi/virat2
Viewer
•
Updated
•
12.8k
•
10
tensorkelechi/repo_id
Viewer
•
Updated
•
12.8k
•
32
tensorkelechi/laion_moondream_caption_10k
Updated
•
31
tensorkelechi/fma_medium
Updated
•
29
tensorkelechi/freesound_mini
Viewer
•
Updated
•
100
•
32
tensorkelechi/freesound_audio_sm
Viewer
•
Updated
•
7.45k
•
154