cnmy-ro / nabla

Minimal autograd engine

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Nabla

Minimal implementation of reverse-mode automatic differentiation.

  • Python version:
    • python/nabla.py: Thin autodiff wrapper over Numpy with PyTorch-like API
    • python_examples: Toy examples built using this module
  • C version (WIP):
    • c/cpuarrays.h : Low-level array library for CPU
    • c/nabla.h: Autodiff library wrapping cpuarrays
    • c_examples: Toy examples built using this library

About

Minimal autograd engine

License:MIT License


Languages

Language:Python 58.4%Language:C 38.8%Language:Jupyter Notebook 2.6%Language:Shell 0.2%