README.md
setup.py
needle/__init__.py
needle/autograd.py
needle/backend_numpy.py
needle/backend_selection.py
needle/data.py
needle/init.py
needle/nn.py
needle/ops.py
needle/optim.py
needle/backend_ndarray/__init__.py
needle/backend_ndarray/ndarray.py
needle/backend_ndarray/ndarray_backend_cpu.cpython-310-x86_64-linux-gnu.so
needle/backend_ndarray/ndarray_backend_cuda.cpython-310-x86_64-linux-gnu.so
needle/backend_ndarray/ndarray_backend_numpy.py
needleTensor.egg-info/PKG-INFO
needleTensor.egg-info/SOURCES.txt
needleTensor.egg-info/dependency_links.txt
needleTensor.egg-info/requires.txt
needleTensor.egg-info/top_level.txt
tests/test_ndarray.py