JAX support
#2
by
MarioGeiger
- opened
Do you have the parameters for the JAX version of MACE? :p
Unfortunately we don't at the moment and I personally have no experience in converting models between jax and pytorch, but I am happy to help :)