Needle

My implementation of Needle：CMU's Deep Learning Systems

Build your own deep learning framework based on C/C++, CUDA, and Python, with GPU support and automatic differentiation

学习文档

Development

配置环境添加 export PYTHONPATH="/root/code/needle:$PYTHONPATH"

vim ~/.bashrc
source ~/.bashrc

在 ipynb 中添加

import sys
sys.path.append("/root/code/needle")

同时需要切换路径，这样才能只执行 needle 下面的 tests，而不受外部干扰

%cd /root/code/needle

TODO

model.to(device) 的实现

RNN/LSTM/Transformer 看情况写不写吧