JAX ports of popular models(audio, vision, llm, diffusion, etc)
kelechic
tensorkelechi
AI & ML interests
vision
Recent Activity
updated
a collection
9 days ago
SAE
updated
a collection
22 days ago
SAE
updated
a collection
22 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
•
47
tensorkelechi/katara-mini
Updated
tensorkelechi/leaf_disease_mobilenet
0.0B
•
Updated
•
5
tensorkelechi/HAR_mobilenet
Image Classification
•
0.0B
•
Updated
•
23
•
1
datasets
13
tensorkelechi/latent_flowers102
Viewer
•
Updated
•
1.02k
•
23
•
1
tensorkelechi/tinywebvid
Viewer
•
Updated
•
8.19k
•
630
tensorkelechi/tiny_webvid_latents
Viewer
•
Updated
•
64
•
34
tensorkelechi/tiny_webvid
Viewer
•
Updated
•
1.02k
•
120
tensorkelechi/virat2
Viewer
•
Updated
•
12.8k
•
13
tensorkelechi/repo_id
Viewer
•
Updated
•
12.8k
•
32
tensorkelechi/laion_moondream_caption_10k
Updated
•
23
tensorkelechi/fma_medium
Updated
•
27
tensorkelechi/freesound_mini
Viewer
•
Updated
•
100
•
21
tensorkelechi/freesound_audio_sm
Viewer
•
Updated
•
7.45k
•
115