physicsnemo
physicsnemo copied to clipboard
🚀[FEA]: Add `from_jax` and `ONNX` support for Models
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