README.md
setup.py
model/__init__.py
model/foundation/RNN.py
model/foundation/Resnet.py
model/foundation/VGG.py
model/foundation/__init__.py
model/llama/__init__.py
model/llama/llama.py
model/llama/test_llama_torch.py
model/transformer/__init__.py
model/transformer/transformer.py
stensor/Config.py
stensor/__init__.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/common/utils.py
stensor/dataset/__init__.py
stensor/dataset/dataloaders.py
stensor/dataset/datasets.py
stensor/dataset/transforms.py
stensor/dataset/utils.py
stensor/nn/__init__.py
stensor/nn/container.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/operations/__init__.py
stensor/ops/operations/_impl.py
stensor/ops/operations/activation_ops.py
stensor/ops/operations/common_ops.py
stensor/ops/operations/math_ops.py
stensor/ops/operations/nn_ops.py
tests/__init__.py
tests/model/__init__.py
tests/model/test_transformer.py
tests/model/test_transformer_torch.py
tests/nn_grad/__init__.py
tests/nn_grad/test_linear.py
tests/nn_grad/test_mnist.py
tests/nn_grad/test_multi_classification.py
tests/nn_grad/test_nonlinear.py
tests/nn_grad/test_vgg.py
tests/ops_grad/__init__.py
tests/ops_grad/test_functional.py
tests/ops_grad/test_grad.py