AI & ML interests

Jax and Flax, high performance computing

StaticFlax 's models

None public yet