vitax. Implementation of a Vision transformer in JAX NNX

github.com/maurock/vitax
by @maurock

Problem

Audience

Keep exploring

Updates

No founder updates yet. Check back soon.