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.