JAX ports of popular models(audio, vision, llm, diffusion, etc)
kelechic
tensorkelechi
AI & ML interests
vision
Recent Activity
liked
a dataset 28 days ago
danjacobellis/musdb18HQ liked
a model about 1 month ago
facebook/sam-audio-small upvoted an article 3 months ago
Continuous batching from first principles Organizations
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
• 30
tensorkelechi/katara-mini
Updated
tensorkelechi/leaf_disease_mobilenet
Updated
• 2
tensorkelechi/HAR_mobilenet
Image Classification • Updated
• 1 • 1
datasets 13
tensorkelechi/latent_flowers102
Viewer
• Updated
• 1.02k • 15 • 1
tensorkelechi/tinywebvid
Viewer
• Updated
• 8.19k • 160
tensorkelechi/tiny_webvid_latents
Viewer
• Updated
• 64 • 12
tensorkelechi/tiny_webvid
Viewer
• Updated
• 1.02k • 68
tensorkelechi/virat2
Viewer
• Updated
• 12.8k • 16
tensorkelechi/repo_id
Viewer
• Updated
• 12.8k • 11
tensorkelechi/laion_moondream_caption_10k
Updated
• 4
tensorkelechi/fma_medium
Updated
• 7
tensorkelechi/freesound_mini
Viewer
• Updated
• 100 • 12
tensorkelechi/freesound_audio_sm
Viewer
• Updated
• 7.45k • 57