CMakeLists.txt
LICENSE
MANIFEST.in
README.md
pyproject.toml
benchmarks/bench_attention.py
benchmarks/bench_conv.py
benchmarks/bench_dataloader.py
benchmarks/bench_dropout.py
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/cpu/kernels/mul.cpp
csrc/backends/cpu/kernels/relu.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
docs/autograd.md
docs/backend-architecture.md
docs/data.md
docs/modules.md
docs/native-backend-roadmap.md
docs/nn.md
docs/randomness.md
docs/releasing.md
docs/serialization.md
docs/tensor-api.md
docs/training.md
examples/checkpoint_training.py
examples/dropout_batchnorm.py
examples/save_load.py
examples/tiny_transformer.py
examples/train_classifier.py
examples/train_cnn_toy.py
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/random.py
python/modelstudio/serialization.py
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/data/__init__.py
python/modelstudio/data/dataloader.py
python/modelstudio/data/dataset.py
python/modelstudio/nn/__init__.py
python/modelstudio/nn/activations.py
python/modelstudio/nn/convolution.py
python/modelstudio/nn/embedding.py
python/modelstudio/nn/init.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/nn/pooling.py
python/modelstudio/nn/transformer.py
python/modelstudio/nn/utils.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
scripts/smoke_test.py
tests/test_attention.py
tests/test_autograd.py
tests/test_batchnorm.py
tests/test_buffers.py
tests/test_concat_stack.py
tests/test_conv.py
tests/test_data.py
tests/test_dataloader_seed.py
tests/test_dispatcher.py
tests/test_dropout.py
tests/test_embedding.py
tests/test_grad_clip.py
tests/test_gradcheck.py
tests/test_indexing.py
tests/test_init.py
tests/test_loss_reductions.py
tests/test_losses.py
tests/test_module_ergonomics.py
tests/test_nn.py
tests/test_norms.py
tests/test_ops.py
tests/test_optim.py
tests/test_optimizer_state.py
tests/test_pooling.py
tests/test_random.py
tests/test_reductions_axis.py
tests/test_serialization.py
tests/test_shape_ops.py
tests/test_state_dict.py
tests/test_tensor.py
tests/test_transformer.py
tests/test_unary_ops.py
tests/test_version.py