add-jax-weights
#1
by
jacobthebanana
- opened
No description provided.
jacobthebanana
changed pull request title from
jax
to add-jax-weights
Converted PyTorch weights to JAX.
jacobthebanana
changed pull request status to
open
Thanks. I will have a try with jax and then merge your work.