docs

README.md

@@ -40,9 +40,9 @@ Or you can just copy `src/slope` to your projects.
 1. Forward-mode, reverse-mode, and higher-order AD.
 
 2. Just-in-time compilation, with interchangeable backends, supporting CPU, CUDA and Metal:
+    - [ONNX Runtime](https://onnxruntime.ai/) (ONNX graph); this is the default backend
     - [IREE](https://iree.dev/) (StableHLO MLIR)
-    - [ONNX Runtime](https://onnxruntime.ai/) (ONNX)
-    - NumPy (CPU-only)
+    - NumPy (Python code)
 
 3. Training and inference, examples:
     - [MLP on MNIST](examples/nn/mnist_mlp.py)

src/slope/__init__.py

@@ -1,27 +1,12 @@
-"""
-.. include:: ../../README.md
-"""
-__all__ = ["core"]
+__all__ = ["core", "nn", "operators", "procedures"]
 __docformat__ = "markdown"
 import os
 from slope import core
-import importlib
 import numpy as np
 np.set_printoptions(precision=5, threshold=1000, edgeitems=5, linewidth=120)
-# SLOPE_BACKEND = os.environ.get("SLOPE_BACKEND", "iree")
 SLOPE_BACKEND = os.environ.get("SLOPE_BACKEND", "onnxruntime")
-# SLOPE_BACKEND = os.environ.get("SLOPE_BACKEND", "numpy")
-# try:
 core.set_backend(SLOPE_BACKEND)
 
-# except Exception as e:
-#     import traceback
-
-#     traceback.print_stack()
-#     print(e)
-#     print(f"\n  -- Warning: failed to set {SLOPE_BACKEND} as backend. Error above \n")
-
-
 def __getattr__(attr):
     if attr in (globals_dict := globals()):
         core.dblog(