Merge branch 'main' into justinchu/ser-initializer

Author: justinchuby

.github/workflows/main.yaml

@@ -99,12 +99,32 @@ jobs:
         with:
           name: Error reports (${{ matrix.name }}-${{ matrix.os }})
           path: error_reports
-      - name: Upload IR profiling results
-        if: matrix.name == 'py311' || matrix.name == 'py311-onnx-weekly'
-        uses: actions/upload-artifact@v3
+
+  dort:
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ubuntu-latest]
+        transformers: ["4.37.2", "4.41.2"]
+        torch: ["release", "nightly"]
+        python_version: ["3.11"]
+        nox-tag: ["test-dort"]
+        name:
+          - dort
+    runs-on: ${{ matrix.os }}
+    steps:
+      - uses: actions/checkout@v4
+      - name: Setup Python ${{ matrix.python_version }}
+        uses: actions/setup-python@v5
         with:
-          name: IR profiling results
-          path: tests/ir/serde_test_profiles
+          python-version: ${{ matrix.python_version }}
+      - name: Install nox
+        run: python -m pip install nox
+      - name: Pull Test Data
+        run: git lfs pull
+      - run: |
+          nox -t ${{ matrix.nox-tag }} --forcecolor -- ${{ matrix.torch }} ${{ matrix.transformers }}
+        name: Run tests
 
   build_docs:
     strategy:

.gitignore

@@ -41,6 +41,7 @@ coverage.xml
 .pytest_cache/
 cover/
 test-output.xml
+*.sarif
 
 # Sphinx documentation
 docs/_build/
@@ -93,6 +94,8 @@ dmypy.json
 
 # Generated files
 *.onnx
+*.csv
+*.xlsx
 !testdata/**/*.onnx
 *.onnxlib
 **/onnx_backend_test_code/**

README.md

@@ -18,6 +18,8 @@ models using a subset of Python. ONNX Script is:
 Note however that ONNX Script does **not** intend to support the entirety
 of the Python language.
 
+Website: [https://onnxscript.ai/](https://onnxscript.ai/)
+
 ## Design Overview
 
 ONNX Script provides a few major capabilities for authoring and debugging

docs/api/index.md

@@ -1,8 +1,17 @@
 # API
 
+## Author Models
+
 ```{toctree}
 decorator
 opsets
 converter
 values
 ```
+
+## Tests and Tools
+
+```{toctree}
+testing
+tools
+```

docs/api/testing.md

@@ -0,0 +1,6 @@
+# Testing
+
+```{eval-rst}
+.. automodule:: onnxscript.testing
+    :members:
+```

docs/api/tools.md

@@ -0,0 +1,19 @@
+# Tools
+
+## Transformers Models
+
+```{eval-rst}
+.. autofunction:: onnxscript.tools.transformers_models.get_model_and_inputs
+```
+
+```{eval-rst}
+.. autofunction:: onnxscript.tools.transformers_models.phi.get_phi_model_from_config
+```
+
+```{eval-rst}
+.. autofunction:: onnxscript.tools.transformers_models.phi3.get_phi3_model_from_config
+```
+
+```{eval-rst}
+.. autofunction:: onnxscript.tools.transformers_models.llama.get_llama_model_from_config
+```

docs/conf.py

@@ -1,5 +1,9 @@
-# Configuration file for the Sphinx documentation builder.
-# To run the documentation: python -m sphinx docs dist/html
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
+"""Configuration file for the Sphinx documentation builder.
+
+To run the documentation: python -m sphinx docs dist/html
+"""
 
 import os
 import re

docs/examples/01_plot_selu.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 Generating a FunctionProto
 ==========================

docs/examples/02_plot_square_loss.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 Generating a ModelProto
 =======================

docs/examples/03_export_lib.py

@@ -1,3 +1,5 @@
+# Copyright (c) Microsoft Corporation.
+# Licensed under the MIT License.
 """
 Generating a LibProto
 =====================