Replies: 1 comment 2 replies
-
|
Beta Was this translation helpful? Give feedback.
2 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
-
dp convert-backend model.pth model.jax
, it outputs a checkpoint folder. How can I load it and perform inference with JAX?DeepPot
relies on NumPy. How can I directly use JAX or PyTorch data types for inference and compute gradients?Thank you so much in advance for your help! I really appreciate it if you could provide guidance.
Beta Was this translation helpful? Give feedback.
All reactions