physicsnemo icon indicating copy to clipboard operation
physicsnemo copied to clipboard

🚀[FEA]: Add `from_jax` and `ONNX` support for Models

Open loliverhennigh opened this issue 2 years ago • 0 comments

Is this a new feature, an improvement, or a change to existing functionality?

New Feature

How would you describe the priority of this feature request

Critical (currently preventing usage)

Please provide a clear description of problem you would like to solve.

Currently we can convert torch.nn.Modulus to modulus modules. Ideally we would have this support for Jax and ONNX models as well.

Describe any alternatives you have considered

No response

loliverhennigh avatar Aug 04 '23 21:08 loliverhennigh