CMakeLists.txt
LICENSE
MANIFEST.in
README.md
pyproject.toml
benchmarks/bench_matmul.py
benchmarks/bench_mlp.py
csrc/CMakeLists.txt
csrc/backends/cpu/cpu_backend.cpp
csrc/backends/cpu/cpu_backend.hpp
csrc/backends/cpu/kernels/add.cpp
csrc/backends/cpu/kernels/matmul.cpp
csrc/backends/cuda/README.md
csrc/backends/cuda/cuda_backend.cu
csrc/backends/cuda/cuda_backend.hpp
csrc/backends/cuda/cuda_memory.hpp
csrc/backends/oneapi/README.md
csrc/backends/oneapi/oneapi_backend.cpp
csrc/backends/oneapi/oneapi_backend.hpp
csrc/backends/oneapi/sycl_memory.hpp
csrc/backends/rocm/README.md
csrc/backends/rocm/hip_memory.hpp
csrc/backends/rocm/rocm_backend.cpp
csrc/backends/rocm/rocm_backend.hpp
csrc/bindings/python_bindings.cpp
csrc/core/device.hpp
csrc/core/dtype.hpp
csrc/core/error.hpp
csrc/core/shape.hpp
csrc/core/storage.hpp
csrc/core/tensor.hpp
csrc/dispatcher/backend.hpp
csrc/dispatcher/dispatcher.hpp
csrc/dispatcher/operator_registry.hpp
examples/train_mlp.py
python/modelstudio/__init__.py
python/modelstudio/_version.py
python/modelstudio/device.py
python/modelstudio/dtypes.py
python/modelstudio/errors.py
python/modelstudio/py.typed
python/modelstudio/storage.py
python/modelstudio/tensor.py
python/modelstudio.egg-info/PKG-INFO
python/modelstudio.egg-info/SOURCES.txt
python/modelstudio.egg-info/dependency_links.txt
python/modelstudio.egg-info/requires.txt
python/modelstudio.egg-info/top_level.txt
python/modelstudio/autograd/__init__.py
python/modelstudio/autograd/engine.py
python/modelstudio/autograd/function.py
python/modelstudio/autograd/grad_mode.py
python/modelstudio/compile/__init__.py
python/modelstudio/compile/graph_capture.py
python/modelstudio/compile/ir.py
python/modelstudio/compile/passes.py
python/modelstudio/nn/__init__.py
python/modelstudio/nn/activations.py
python/modelstudio/nn/linear.py
python/modelstudio/nn/losses.py
python/modelstudio/nn/module.py
python/modelstudio/nn/normalization.py
python/modelstudio/nn/parameter.py
python/modelstudio/ops/__init__.py
python/modelstudio/ops/creation.py
python/modelstudio/ops/linalg.py
python/modelstudio/ops/math.py
python/modelstudio/ops/movement.py
python/modelstudio/ops/reductions.py
python/modelstudio/optim/__init__.py
python/modelstudio/optim/adamw.py
python/modelstudio/optim/optimizer.py
python/modelstudio/optim/sgd.py
python/modelstudio/runtime/__init__.py
python/modelstudio/runtime/backend.py
python/modelstudio/runtime/dispatcher.py
python/modelstudio/testing/__init__.py
python/modelstudio/testing/gradcheck.py
tests/test_autograd.py
tests/test_dispatcher.py
tests/test_gradcheck.py
tests/test_nn.py
tests/test_ops.py
tests/test_optim.py
tests/test_tensor.py
tests/test_version.py