Jaxonnxruntime

Latest version: v0.3.0

Safety actively analyzes 623827 Python packages for vulnerabilities to keep your Python projects secure.

Scan your dependencies

0.3.0

1. Create a call_torch API to make Pytorch a frontend of JAX. The call_torch API achieves two tasks:
(1) Convert the torch nn.Module or torch scriptmodel to onnx format by torch.onnx.export
(2) Use jaxonnxruntime to convert the ONNX model into JAX function
We hook all necessary components to make life easier for end PyTorch users. The input format should follow the torch.onnx.export API.

2. Improve the tests to cover more corner cases and be more robust

3. Fix more ops to support more models and tests from ONNX backend test suite

0.2.0

Changes includes:
1. Add more onnx ops and onnx model zoo model (42) support.
2. Run through Meta LLAMA model end 2 end flow.
3. Add doc how to contribute new op.

0.1.0

This version passes all 'real' model tests provided by onnx.

Links

Releases

Has known vulnerabilities

© 2024 Safety CLI Cybersecurity Inc. All Rights Reserved.