AI & ML interests

Jax and Flax, high performance computing

StaticFlax 's datasets

None public yet