CMakeLists.txt
LICENSE
MANIFEST.in
README.md
pyproject.toml
benchmarks/bench_attention.py
benchmarks/bench_conv.py
benchmarks/bench_creation.py
benchmarks/bench_cuda_autograd.py
benchmarks/bench_cuda_elementwise.py
benchmarks/bench_cuda_matmul.py
benchmarks/bench_dataloader.py
benchmarks/bench_dropout.py
benchmarks/bench_elementwise.py
benchmarks/bench_manipulation.py
benchmarks/bench_matmul.py
benchmarks/bench_mlp.py
benchmarks/bench_trace.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_context.cu
csrc/backends/cuda/cuda_context.hpp
csrc/backends/cuda/cuda_kernels.hpp
csrc/backends/cuda/cuda_memory.cu
csrc/backends/cuda/cuda_memory.hpp
csrc/backends/cuda/cuda_stream.cu
csrc/backends/cuda/cuda_stream.hpp
csrc/backends/cuda/kernels/elementwise.cu
csrc/backends/cuda/kernels/matmul.cu
csrc/backends/cuda/kernels/reductions.cu
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/cuda_bindings.cpp
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/backend-status.md
docs/checkpointing.md
docs/comparison-ops.md
docs/cuda.md
docs/data.md
docs/functional-api.md
docs/linalg.md
docs/metrics.md
docs/modules.md
docs/native-backend-roadmap.md
docs/nn.md
docs/numpy-interop.md
docs/optimizers.md
docs/random.md
docs/randomness.md
docs/releasing.md
docs/serialization.md
docs/tensor-api.md
docs/tensor-creation.md
docs/tensor-manipulation.md
docs/tracing.md
docs/training.md
examples/backend_status.py
examples/checkpoint_resume.py
examples/checkpoint_training.py
examples/cuda_autograd_demo.py
examples/cuda_mlp_demo.py
examples/cuda_tensor_demo.py
examples/dropout_batchnorm.py
examples/functional_training.py
examples/metrics_demo.py
examples/numpy_interop.py
examples/random_linalg_demo.py
examples/save_load.py
examples/scheduler_training.py
examples/tiny_transformer.py
examples/tracing_demo.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/interop.py
python/modelstudio/linalg.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/backends/__init__.py
python/modelstudio/backends/cuda.py
python/modelstudio/backends/status.py
python/modelstudio/compile/__init__.py
python/modelstudio/compile/graph_capture.py
python/modelstudio/compile/ir.py
python/modelstudio/compile/passes.py
python/modelstudio/cuda/__init__.py
python/modelstudio/cuda/device.py
python/modelstudio/cuda/memory.py
python/modelstudio/cuda/streams.py
python/modelstudio/data/__init__.py
python/modelstudio/data/dataloader.py
python/modelstudio/data/dataset.py
python/modelstudio/metrics/__init__.py
python/modelstudio/metrics/classification.py
python/modelstudio/nn/__init__.py
python/modelstudio/nn/activations.py
python/modelstudio/nn/convolution.py
python/modelstudio/nn/embedding.py
python/modelstudio/nn/functional.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/comparison.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/lr_scheduler.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/cuda_release_check.py
scripts/cuda_source_build_check.py
scripts/smoke_test.py
tests/test_activations_more.py
tests/test_attention.py
tests/test_autograd.py
tests/test_backend_status.py
tests/test_batchnorm.py
tests/test_buffers.py
tests/test_checkpoint_helpers.py
tests/test_clone_copy.py
tests/test_comparison_ops.py
tests/test_concat_stack.py
tests/test_conv.py
tests/test_creation_more.py
tests/test_cuda_autograd.py
tests/test_cuda_availability.py
tests/test_cuda_matmul.py
tests/test_cuda_memory.py
tests/test_cuda_nn.py
tests/test_cuda_ops.py
tests/test_cuda_optim.py
tests/test_cuda_reductions.py
tests/test_cuda_tensor.py
tests/test_data.py
tests/test_data_split.py
tests/test_dataloader_seed.py
tests/test_dispatcher.py
tests/test_dropout.py
tests/test_dtype_conversion.py
tests/test_embedding.py
tests/test_functional.py
tests/test_grad_clip.py
tests/test_gradcheck.py
tests/test_indexing.py
tests/test_indexing_assignment.py
tests/test_init.py
tests/test_linalg.py
tests/test_loss_reductions.py
tests/test_losses.py
tests/test_lr_scheduler.py
tests/test_manipulation_ops.py
tests/test_metrics.py
tests/test_module_ergonomics.py
tests/test_native_cpu_mode.py
tests/test_nn.py
tests/test_norms.py
tests/test_numpy_interop.py
tests/test_ops.py
tests/test_optim.py
tests/test_optimizer_param_groups.py
tests/test_optimizer_state.py
tests/test_pooling.py
tests/test_public_exports.py
tests/test_random.py
tests/test_random_namespace.py
tests/test_reductions_axis.py
tests/test_scalar_behavior.py
tests/test_serialization.py
tests/test_serialization_hardening.py
tests/test_shape_ops.py
tests/test_state_dict.py
tests/test_tensor.py
tests/test_trace.py
tests/test_transformer.py
tests/test_unary_ops.py
tests/test_version.py