JAX ports of popular models(audio, vision, llm, diffusion, etc)
kelechic
tensorkelechi
AI & ML interests
vision
Recent Activity
liked
a dataset
about 15 hours ago
danjacobellis/musdb18HQ
liked
a model
4 days ago
facebook/sam-audio-small
upvoted
an
article
about 2 months ago
Continuous batching from first principles