LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
include/arena.h
include/arena_cpu.h
include/arena_cuda.h
include/definitions.h
include/graph.h
include/node.h
include/op.h
include/ops.def
include/tensor.h
include/kernels/abs.h
include/kernels/add.h
include/kernels/broadcast.h
include/kernels/conv2d.h
include/kernels/cos.h
include/kernels/cpu_utils.h
include/kernels/div.h
include/kernels/exp.h
include/kernels/expand.h
include/kernels/flatten.h
include/kernels/iter.h
include/kernels/leaky_relu.h
include/kernels/log.h
include/kernels/matmul.h
include/kernels/max.h
include/kernels/mean.h
include/kernels/min.h
include/kernels/mul.h
include/kernels/neg.h
include/kernels/pack.h
include/kernels/sin.h
include/kernels/squeeze.h
include/kernels/sub.h
include/kernels/sum.h
include/kernels/tan.h
include/kernels/transpose.h
include/kernels/unsqueeze.h
include/kernels/view.h
include/kernels/cpu/cpu_tensor_init.h
include/kernels/cuda/cuda_check.h
include/kernels/cuda/cuda_iter.cuh
include/kernels/cuda/cuda_tensor_init.h
include/kernels/cuda/cuda_utils.cuh
include/optimizers/adam.h
include/optimizers/adamw.h
include/optimizers/sgd.h
include/optimizers/zero_grad.h
plast/__init__.py
plast/_internal.py
plast/tensor_ops.py
pyplast.egg-info/PKG-INFO
pyplast.egg-info/SOURCES.txt
pyplast.egg-info/dependency_links.txt
pyplast.egg-info/requires.txt
pyplast.egg-info/top_level.txt
src/arena.c
src/arena_cpu.c
src/arena_cuda.cu
src/graph.c
src/node.c
src/op.c
src/tensor.c
src/kernels/cpu/abs.c
src/kernels/cpu/add.c
src/kernels/cpu/broadcast.c
src/kernels/cpu/conv2d.c
src/kernels/cpu/cos.c
src/kernels/cpu/cpu_tensor_init.c
src/kernels/cpu/cpu_utils.c
src/kernels/cpu/div.c
src/kernels/cpu/exp.c
src/kernels/cpu/expand.c
src/kernels/cpu/flatten.c
src/kernels/cpu/iter.c
src/kernels/cpu/leaky_relu.c
src/kernels/cpu/log.c
src/kernels/cpu/matmul.c
src/kernels/cpu/max.c
src/kernels/cpu/mean.c
src/kernels/cpu/min.c
src/kernels/cpu/mul.c
src/kernels/cpu/neg.c
src/kernels/cpu/pack.c
src/kernels/cpu/sin.c
src/kernels/cpu/squeeze.c
src/kernels/cpu/sub.c
src/kernels/cpu/sum.c
src/kernels/cpu/tan.c
src/kernels/cpu/transpose.c
src/kernels/cpu/unsqueeze.c
src/kernels/cpu/view.c
src/kernels/cuda/abs.cu
src/kernels/cuda/add.cu
src/kernels/cuda/conv2d.cu
src/kernels/cuda/cos.cu
src/kernels/cuda/cuda_iter_init.cu
src/kernels/cuda/cuda_pack.cu
src/kernels/cuda/cuda_tensor_init.cu
src/kernels/cuda/div.cu
src/kernels/cuda/exp.cu
src/kernels/cuda/leaky_relu.cu
src/kernels/cuda/log.cu
src/kernels/cuda/matmul.cu
src/kernels/cuda/max.cu
src/kernels/cuda/mean.cu
src/kernels/cuda/min.cu
src/kernels/cuda/mul.cu
src/kernels/cuda/neg.cu
src/kernels/cuda/sin.cu
src/kernels/cuda/sub.cu
src/kernels/cuda/sum.cu
src/kernels/cuda/tan.cu
src/optimizers/cpu/adam.c
src/optimizers/cpu/adamw.c
src/optimizers/cpu/sgd.c
src/optimizers/cpu/zero_grad.c
src/optimizers/cuda/adam.cu
src/optimizers/cuda/adamw.cu
src/optimizers/cuda/sgd.cu
src/optimizers/cuda/zero_grad.cu
src/python/bindings.cpp
tests/test_data.py
tests/test_e2e.py
tests/test_edge_cases.py
tests/test_experiment.py
tests/test_gradients.py
tests/test_nn.py
tests/test_ops.py
tests/test_optim.py
tests/test_tensor.py