You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I want to ask if it is possible to use Jax / Flax etc. libraries instead of PyTorch to close a complete learning pipeline. Jax offers just-in-time compiling and auto-diff for native python functions and it would be nice to use it with the differentiation feature of genesis. Has somebody maybe tried that already?
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
I want to ask if it is possible to use Jax / Flax etc. libraries instead of PyTorch to close a complete learning pipeline. Jax offers just-in-time compiling and auto-diff for native python functions and it would be nice to use it with the differentiation feature of genesis. Has somebody maybe tried that already?
Beta Was this translation helpful? Give feedback.
All reactions