LICENSE
README.md
setup.py
stensor/__init__.py
stensor/config.py
stensor.egg-info/PKG-INFO
stensor.egg-info/SOURCES.txt
stensor.egg-info/dependency_links.txt
stensor.egg-info/top_level.txt
stensor/common/__init__.py
stensor/common/_register_for_tensor.py
stensor/common/tensor.py
stensor/dataset/__init__.py
stensor/dataset/dataloaders.py
stensor/dataset/datasets.py
stensor/dataset/transforms.py
stensor/dataset/utils.py
stensor/model/__init__.py
stensor/model/foundation/RNN.py
stensor/model/foundation/Resnet.py
stensor/model/foundation/VGG.py
stensor/model/foundation/__init__.py
stensor/model/gpt/__init__.py
stensor/model/gpt/dataset.py
stensor/model/gpt/gpt_config.py
stensor/model/gpt/gpt_model.py
stensor/model/gpt/rope.py
stensor/model/llama/__init__.py
stensor/model/llama/llama.py
stensor/model/transformer/__init__.py
stensor/model/transformer/transformer.py
stensor/nn/__init__.py
stensor/nn/container.py
stensor/nn/hook.py
stensor/nn/metric.py
stensor/nn/module.py
stensor/nn/utils.py
stensor/nn/layer/__init__.py
stensor/nn/layer/activation.py
stensor/nn/layer/convolution.py
stensor/nn/layer/embedding.py
stensor/nn/layer/linear.py
stensor/nn/layer/normalization.py
stensor/nn/layer/pooling.py
stensor/nn/layer/rnn.py
stensor/nn/loss/__init__.py
stensor/nn/loss/loss.py
stensor/nn/opt/__init__.py
stensor/nn/opt/optimizer.py
stensor/ops/__init__.py
stensor/ops/_type_check.py
stensor/ops/functional.py
stensor/ops/primitive.py
stensor/ops/kernel/__init__.py
stensor/ops/kernel/dispatcher.py
stensor/ops/kernel/aclop/__init__.py
stensor/ops/kernel/aclop/aclop_impl.py
stensor/ops/kernel/aclop/aclop_kernel.py
stensor/ops/kernel/aclop/activation_ops.py
stensor/ops/kernel/aclop/common_ops.py
stensor/ops/kernel/aclop/math_ops.py
stensor/ops/kernel/aclop/nn_ops.py
stensor/ops/kernel/cupy/__init__.py
stensor/ops/kernel/cupy/_impl.py
stensor/ops/kernel/cupy/activation_ops.py
stensor/ops/kernel/cupy/common_ops.py
stensor/ops/kernel/cupy/cupy_kernel.py
stensor/ops/kernel/cupy/infer_ops.py
stensor/ops/kernel/cupy/math_ops.py
stensor/ops/kernel/cupy/nn_ops.py
stensor/ops/kernel/numpy/__init__.py
stensor/ops/kernel/numpy/_impl.py
stensor/ops/kernel/numpy/activation_ops.py
stensor/ops/kernel/numpy/common_ops.py
stensor/ops/kernel/numpy/infer_ops.py
stensor/ops/kernel/numpy/math_ops.py
stensor/ops/kernel/numpy/nn_ops.py
stensor/ops/kernel/numpy/numpy_kernel.py
tests/__init__.py
tests/nn/__init__.py
tests/nn/test_gpt_model.py
tests/nn/test_linear_regression.py
tests/nn/test_llama.py
tests/nn/test_mlp_with_mnist.py
tests/nn/test_multi_classification.py
tests/nn/test_nonlinear_regression.py
tests/nn/test_transformer.py
tests/nn/test_vgg.py
tests/ops/__init__.py
tests/ops/test_functional.py
tests/ops/test_performance.py