LICENSE
MANIFEST.in
README.md
cnn_demo.py
essentials_demo.py
gpu_demo.py
main.py
pyproject.toml
torch_bridge_demo.py
rasptorch/__init__.py
rasptorch/data.py
rasptorch/functional.py
rasptorch/gpu_training.py
rasptorch/nn.py
rasptorch/optim.py
rasptorch/tensor.py
rasptorch/torch_bridge.py
rasptorch/train.py
rasptorch/vulkan_backend.py
rasptorch.egg-info/PKG-INFO
rasptorch.egg-info/SOURCES.txt
rasptorch.egg-info/dependency_links.txt
rasptorch.egg-info/requires.txt
rasptorch.egg-info/top_level.txt
rasptorch/shaders/add.comp
rasptorch/shaders/add.spv
rasptorch/shaders/add_rowvec.comp
rasptorch/shaders/add_rowvec.spv
rasptorch/shaders/add_scalar.comp
rasptorch/shaders/add_scalar.spv
rasptorch/shaders/col2im_nchw.comp
rasptorch/shaders/col2im_nchw.spv
rasptorch/shaders/im2col_nchw.comp
rasptorch/shaders/im2col_nchw.spv
rasptorch/shaders/layernorm2d.comp
rasptorch/shaders/layernorm2d.spv
rasptorch/shaders/layernorm2d_backward.comp
rasptorch/shaders/layernorm2d_backward.spv
rasptorch/shaders/log_softmax2d.comp
rasptorch/shaders/log_softmax2d.spv
rasptorch/shaders/log_softmax2d_backward.comp
rasptorch/shaders/log_softmax2d_backward.spv
rasptorch/shaders/mat2nchw.comp
rasptorch/shaders/mat2nchw.spv
rasptorch/shaders/matmul.comp
rasptorch/shaders/matmul.spv
rasptorch/shaders/matmul_a_bt.comp
rasptorch/shaders/matmul_a_bt.spv
rasptorch/shaders/matmul_at_b.comp
rasptorch/shaders/matmul_at_b.spv
rasptorch/shaders/mse_grad.comp
rasptorch/shaders/mse_grad.spv
rasptorch/shaders/mul.comp
rasptorch/shaders/mul.spv
rasptorch/shaders/mul_add_relu.comp
rasptorch/shaders/mul_add_relu.spv
rasptorch/shaders/mul_rowvec.comp
rasptorch/shaders/mul_rowvec.spv
rasptorch/shaders/mul_scalar.comp
rasptorch/shaders/mul_scalar.spv
rasptorch/shaders/nchw2mat.comp
rasptorch/shaders/nchw2mat.spv
rasptorch/shaders/neg.comp
rasptorch/shaders/neg.spv
rasptorch/shaders/reduce_sum_rows.comp
rasptorch/shaders/reduce_sum_rows.spv
rasptorch/shaders/reduce_sum_stage.comp
rasptorch/shaders/reduce_sum_stage.spv
rasptorch/shaders/relu.comp
rasptorch/shaders/relu.spv
rasptorch/shaders/relu_backward.comp
rasptorch/shaders/relu_backward.spv
rasptorch/shaders/scale_fill.comp
rasptorch/shaders/scale_fill.spv
rasptorch/shaders/sgd_momentum_update.comp
rasptorch/shaders/sgd_momentum_update.spv
rasptorch/shaders/sgd_update.comp
rasptorch/shaders/sgd_update.spv
rasptorch/shaders/softmax2d.comp
rasptorch/shaders/softmax2d.spv
rasptorch/shaders/softmax2d_backward.comp
rasptorch/shaders/softmax2d_backward.spv
rasptorch/shaders/softmax_xent_backward.comp
rasptorch/shaders/softmax_xent_backward.spv
rasptorch/shaders/softmax_xent_loss_vec.comp
rasptorch/shaders/softmax_xent_loss_vec.spv
rasptorch/shaders/transpose2d.comp
rasptorch/shaders/transpose2d.spv
tests/test_autograd_cpu.py
tests/test_gpu_autograd_essentials.py
tests/test_nn_essentials.py
tests/test_no_grad.py
tests/test_readme_options_extras.py
tests/test_tensor_ops_cpu.py
tests/test_train_loop_gpu.py
tests/test_vulkan_smoke.py