https://github.com/vaithak/needle

A basic automatic differentiation library and a minimalist layer of neural network modules.
https://github.com/vaithak/needle

Last synced: 8 months ago
JSON representation

A basic automatic differentiation library and a minimalist layer of neural network modules.

• Host: GitHub
• URL: https://github.com/vaithak/needle
• Owner: vaithak
• License: mit
• Created: 2023-03-12T18:47:34.000Z (over 2 years ago)
• Default Branch: main
• Last Pushed: 2023-04-01T06:29:05.000Z (over 2 years ago)
• Last Synced: 2025-01-22T07:45:40.248Z (9 months ago)
• Language: Python
• Homepage:
• Size: 224 KB
• Stars: 0
• Watchers: 1
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

          
# needle

![image](needle.png)  

**NE**cessary **E**lements of **D**eep **LE**arning

A basic autograd engine with a minimalist module for implementing neural nets above it and GPU-supported array backend.  

It works by doing reverse mode AD (Automatic Differentiation) over the dynamically built computational graph.

This was a personal just for fun and learning project built using course assignments for [CMU's DL systems course](https://dlsyscourse.org/)