ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator
Find a file
Abhishek Jindal 91c940b619
adding fill scalar for torch ones direct initialization on ort device (#10898)
* adding fill scalar for torch ones direct initialization on device and adding test case for it

* using ConstantOfShape to for implementing fill Scalar in atenops

* adding case for handling at::Tensor attribute

* handling the at::Tensor type for ConstantOfShape

* handling the at::Tensor type for ConstantOfShape with attr type

* handling the at::Tensor type case

* converting the data to tensor in case of aten tensor mapping is needed

* handling aten tensor case

* handling aten tensor case and reversing the string case

* changing type of scalar
2022-04-05 11:17:25 -07:00
.config A new pipeline to replace the existing WindowsAI packaging pipeline (#10646) 2022-03-03 08:56:49 -08:00
.gdn
.github Refactor Python API docs to better explain IO binding scenarios (#10651) 2022-03-15 09:40:59 -07:00
.pipelines A new pipeline to replace the existing WindowsAI packaging pipeline (#10646) 2022-03-03 08:56:49 -08:00
cgmanifests [TVM EP] code refactor (#10655) 2022-03-16 13:55:04 +01:00
cmake Fix ARM64EC build breaks (#11111) 2022-04-05 10:00:42 -07:00
csharp [CUDA] Optimize Conv and ConvGrad for Training (#10999) 2022-03-29 07:31:36 +08:00
dockerfiles Update rocm_ep and migraphx_ep to rocm4.5.2 and fix dockerfiles to build docker images correctly (#10445) 2022-02-01 16:11:39 -08:00
docs update How_To_Update_ONNX_Dev_Notes with right paths (#11074) 2022-04-01 08:05:31 -07:00
include/onnxruntime/core [CUDA] Optimize Conv and ConvGrad for Training (#10999) 2022-03-29 07:31:36 +08:00
java [Java] Support configuring CUDA and TensorRT execution providers (#10697) 2022-03-30 14:26:51 -07:00
js [js] aggregation of recent dependabot security warnings fix (#11060) 2022-03-31 02:06:04 -07:00
objectivec [iOS packaging] Minor updates. (#10755) 2022-03-04 16:02:53 +10:00
onnxruntime Add function body for SoftmaxCrossEntropyLossGrad (#10779) 2022-04-05 10:52:40 -07:00
orttraining adding fill scalar for torch ones direct initialization on ort device (#10898) 2022-04-05 11:17:25 -07:00
package/rpm Bump master version to 1.12 (#10797) 2022-03-28 12:30:11 -07:00
samples
server [TVM EP] Rename Standalone TVM (STVM) Execution Provider to TVM EP (#10260) 2022-02-15 10:21:02 +01:00
tools orttraining cuda 10.2 to not build for compute_80 (#11103) 2022-04-04 17:22:05 -07:00
winml Add multi-dim dft test, and fix complex idft (#10947) 2022-03-22 10:08:12 -07:00
.clang-format
.clang-tidy
.dockerignore
.flake8
.gitattributes
.gitignore Remove unused pipeline orttraining-linux-gpu-perf-test-ci-pipeline.yml and unused send_perf_metrics tool. (#10326) 2022-01-21 14:31:34 -08:00
.gitmodules Upgrade emsdk to 3.1.3 (#10577) 2022-02-28 23:52:41 -08:00
build.amd64.1411.bat
build.bat
build.sh
CITATION.cff Add citation file (#10061) 2021-12-16 19:56:21 -08:00
CODEOWNERS Update CODEOWNERS (#10932) 2022-03-18 09:37:58 -07:00
CONTRIBUTING.md fixed the link (#8757) 2021-08-18 11:45:42 -07:00
LICENSE
NuGet.config
ort.wprp
ORT_icon_for_light_bg.png Update nuget icon (#10672) 2022-03-01 09:11:03 -08:00
packages.config Bump winrt version (#10243) 2022-01-12 10:52:27 -08:00
README.md
requirements-dev.txt
requirements-doc.txt
requirements-training.txt
requirements.txt.in
setup.py [TVM EP] Integrate tests for TVM EP into public onnxruntime CI (#10505) 2022-02-24 16:24:23 +01:00
ThirdPartyNotices.txt add copyright (#9943) (#9970) 2021-12-08 14:34:53 -08:00
VERSION_NUMBER Bump master version to 1.12 (#10797) 2022-03-28 12:30:11 -07:00

ONNX Runtime is a cross-platform inference and training machine-learning accelerator.

ONNX Runtime inference can enable faster customer experiences and lower costs, supporting models from deep learning frameworks such as PyTorch and TensorFlow/Keras as well as classical machine learning libraries such as scikit-learn, LightGBM, XGBoost, etc. ONNX Runtime is compatible with different hardware, drivers, and operating systems, and provides optimal performance by leveraging hardware accelerators where applicable alongside graph optimizations and transforms. Learn more →

ONNX Runtime training can accelerate the model training time on multi-node NVIDIA GPUs for transformer models with a one-line addition for existing PyTorch training scripts. Learn more →

Get Started

General Information: onnxruntime.ai

Usage documention and tutorials: onnxruntime.ai/docs

Companion sample repositories:

Build Pipeline Status

System CPU GPU EPs
Windows Build Status Build Status Build Status
Linux Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Mac Build Status
Build Status
Android Build Status
iOS Build Status
WebAssembly Build Status

Data/Telemetry

Windows distributions of this project may collect usage data and send it to Microsoft to help improve our products and services. See the privacy statement for more details.

Contributions and Feedback

We welcome contributions! Please see the contribution guidelines.

For feature requests or bug reports, please file a GitHub Issue.

For general discussion or questions, please use GitHub Discussions.

Code of Conduct

This project has adopted the Microsoft Open Source Code of Conduct. For more information see the Code of Conduct FAQ or contact opencode@microsoft.com with any additional questions or comments.

License

This project is licensed under the MIT License.