LICENSE
MANIFEST.in
README.md
pyproject.toml
setup.py
include/core/arena.h
include/core/arena_cpu.h
include/core/arena_cuda.h
include/core/definitions.h
include/core/graph.h
include/core/node.h
include/core/op.h
include/core/ops.def
include/core/tensor.h
include/data/dataloader.h
include/data/dataset.h
include/kernels/conv2d.h
include/kernels/cpu_utils.h
include/kernels/kernel_macros.h
include/kernels/kernel_types.h
include/kernels/matmul.h
include/kernels/pack.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/kernels/ops/binary.h
include/kernels/ops/reduce.h
include/kernels/ops/shape.h
include/kernels/ops/unary.h
include/lr_scheduler/lr_scheduler.h
include/optimizers/adam.h
include/optimizers/adamw.h
include/optimizers/sgd.h
include/optimizers/zero_grad.h
include/plast/loss.h
include/plast/model.h
include/plast/optim.h
include/plast/plast.h
include/scheduler/fusion.h
include/scheduler/jit.h
include/scheduler/scheduler.h
plast/__init__.py
plast/_internal.py
plast/tensor.py
plast/data/__init__.py
plast/data/dataloader.py
plast/data/dataset.py
plast/experiment/__init__.py
plast/experiment/config.py
plast/experiment/tracker.py
plast/experiment/yaml_utils.py
plast/nn/__init__.py
plast/nn/activation.py
plast/nn/container.py
plast/nn/conv.py
plast/nn/dropout.py
plast/nn/functional.py
plast/nn/linear.py
plast/nn/loss.py
plast/nn/module.py
plast/nn/normalization.py
plast/optim/__init__.py
plast/optim/adam.py
plast/optim/adamw.py
plast/optim/lr_scheduler.py
plast/optim/optimizer.py
plast/optim/sgd.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/allocator.zig
src/c_api.zig
src/device.zig
src/function.zig
src/graph.zig
src/op.zig
src/root.zig
src/tensor.zig
src/uop.zig
src/api/loss.c
src/api/model.c
src/api/optim.c
src/api/serialization.c
src/core/arena.c
src/core/arena_cpu.c
src/core/arena_cuda.cu
src/core/graph.c
src/core/node.c
src/core/op.c
src/core/tensor.c
src/data/dataset.c
src/kernels/cpu/binary_ops.c
src/kernels/cpu/conv2d.c
src/kernels/cpu/cpu_tensor_init.c
src/kernels/cpu/cpu_utils.c
src/kernels/cpu/fused_ops.c
src/kernels/cpu/matmul.c
src/kernels/cpu/pack.c
src/kernels/cpu/reduce_ops.c
src/kernels/cpu/shape_ops.c
src/kernels/cpu/unary_ops.c
src/kernels/cuda/binary_ops.cu
src/kernels/cuda/conv2d.cu
src/kernels/cuda/conv_relu.cu
src/kernels/cuda/cuda_iter_init.cu
src/kernels/cuda/cuda_pack.cu
src/kernels/cuda/cuda_tensor_init.cu
src/kernels/cuda/matmul.cu
src/kernels/cuda/matmul_bias_relu.cu
src/kernels/cuda/matmul_relu.cu
src/kernels/cuda/reduce_ops.cu
src/kernels/cuda/shape_ops_cuda.cu
src/kernels/cuda/unary_ops.cu
src/lr_scheduler/lr_scheduler.c
src/ops/element_wise.zig
src/ops/functions.zig
src/ops/fused.zig
src/ops/linear.zig
src/ops/mod.zig
src/ops/reduce.zig
src/optimizer/mod.zig
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
src/scheduler/fusion.c
src/scheduler/jit.c
src/scheduler/scheduler.c
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