-
Notifications
You must be signed in to change notification settings - Fork 2
Open
Description
- Simple tensor class
- Layers: dense, relu, sigmoid, softmax
- Model: predict, fit (validation), evaluate, summary
- Loss: MSE, cross-entropy
- Metrics: loss, accuracy
- Weights regularization
- Learning rate schedulers
- Optim: Adam, SGD
- Python bindings
- Export/import weights
- Optmizations: CUDA, OpenMP
- Clean CPP, cmake, tests, docs, example notebooks
Objective: fast and accurate classification of MNIST.
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
No labels