Needle - A Minimal Deep Learning Framework

Built a PyTorch-like deep learning framework from scratch

  • Built a PyTorch-like deep learning framework from scratch, implementing dynamic computation graphs and reverse-mode automatic differentiation.
  • Supported core neural architectures (CNNs, RNNs, Transformers), optimizers, and GPU acceleration for end-to-end training.