JAX ports of popular models(audio, vision, llm, diffusion, etc)
kelechic
tensorkelechi
AI & ML interests
vision
Recent Activity
updated
a model
about 1 hour ago
Audiogen/virgil-lyrics-130126
published
a model
about 1 hour ago
Audiogen/virgil-lyrics-130126
upvoted
an
article
about 1 month ago
Continuous batching from first principles