MANIFEST.in
README.md
pyproject.toml
setup.py
csrc/autograd.c
csrc/autograd.h
csrc/dispatch.c
csrc/dispatch.h
csrc/tensor.c
csrc/tensor.h
csrc/tensor_bindings.c
csrc/util.c
csrc/util.h
csrc/cpu/activation_cpu.c
csrc/cpu/matmul_cpu.c
csrc/cpu/norm_cpu.c
csrc/cpu/tensor_cpu.c
csrc/cpu/tensor_cpu.h
csrc/cuda_driver/activation_driver.c
csrc/cuda_driver/cuda_init.c
csrc/cuda_driver/cuda_init.h
csrc/cuda_driver/matmul_driver.c
csrc/cuda_driver/norm_driver.c
csrc/cuda_driver/tensor_cuda.c
csrc/cuda_driver/tensor_cuda.h
src/mx/__init__.py
src/mx/functional.py
src/mx/linear.py
src/mx/module.py
src/mx/norm.py
src/mx/optimizers.py
src/mx/parameter.py
src/mx/tensor.py
src/qkmx.egg-info/PKG-INFO
src/qkmx.egg-info/SOURCES.txt
src/qkmx.egg-info/dependency_links.txt
src/qkmx.egg-info/top_level.txt
test/test.py
test/test_autograd.py
test/test_cuda_ops.py