okbalefthanded/aawedha

[Feature] Add JAX support

Opened this issue · 1 comments

okbalefthanded commented
  • Plenty of High-Level API for JAX exists, we think that Elegy is a good fit due to its Keras like design.
okbalefthanded commented

JAX supported is postponed until a stable windows release will be available.