/pequegrad

Tensors with automatic differentiation and JIT compilation for Python. Pequegrad is a toy deep learning and scientific computing library that runs on the CPU and GPU

Primary LanguageC++MIT LicenseMIT

Watchers