A lightweight educational deep learning library
Project description
QuackNet
Built from scratch by a 16-year-old, QuackNet implements NN, CNN, RNN, and Transformers entirely in NumPy, with 150+ unit tests, PyPI distribution, and benchmarks that outperform PyTorch/TensorFlow on MNIST.
QuackNet is a from scratch deep learning library in NumPy with full support for Neural Networks, CNNs, RNNs, and Transformers. Educational and benchmarked against PyTorch and TensorFlow.
Why QuackNet?
Most high level ML libraries (TensorFlow and PyTorch) abstract away the inner workings of deep learning models.
QuackNet exposes every step of the process:
- Forward and backward propagation
- Manual gradient computation
- Weight and bias updates layer by layer
Built to be both an educational tool and a lightweight research framework, QuackNet allows you to experiment with novel architectures while understanding every underlying computation.
Key Features
- No ML frameworks used: built only with NumPy
- Core Models:
- Neural Network (Dense)
- Convolutional Neural Network (Conv1D/2D, Pooling, Flatten, Global Average Pooling)
- Recurrent Neural Network (Singular and Stacked RNN)
- Transformer (Encoder & Decoder, multi head attention, normalisation, residual connections, positional encoding)
- Activation Functions: ReLU, Leaky ReLU, Sigmoid, SoftMax, Tanh
- Loss Functions: Cross Entropy, MSE, MAE
- Optimisers: GD, SGD, Adam, AdamW, RMSProp, Lion
- Utilities:
- Save/load weights and biases
- Visualise training progress (accuracy/loss graphs)
- Evaluate metrics (accuracy, loss)
- Real world demo projects (MNIST, HAM10000 skin lesions)
- Data augmentation (flipping)
- Professional Engineering:
- PyPi installable
- Automated documentation via
pdoc - 150+ unit tests with 91% coverage
- Modular, beginner friendly API
Installation
Install QuackNet from PyPI:
pip install QuackNet
Quick Start
from quacknet import Network
n = Network(lossFunc="Cross Entropy", learningRate=0.01, optimisationFunc="SGD")
n.addLayer(3)
n.addLayer(1, "SoftMax")
n.createWeightsAndBiases()
inputData = [[0.1, 0.2, 0.3], [0.4, 0.5, 0.6]]
labels = [[1], [0]]
accuracy, averageLoss = n.train(inputData, labels, epochs=5)
For more detailed examples, see Advanced Usage or Examples
Examples
- Simple Neural Network Example: A basic neural network implementation demonstrating forward and backpropagation
- Convolutional Neural Network Example: Shows how to use the convolutional layers in the library
- MNIST Neural Network Example: Trains a neural network on the MNIST dataset using QuackNet
- Singular Recurrent Neural Network Example: Shows how to use the Singular RNN
- Stacked Recurrent Neural Network Example: Shows how to use the Stacked RNN
- Transformer Example: Shows how to use the Transformer
Advanced Usage
Here is an example of how to create and train a simple neural network using the library:
from quacknet import Network
# Define a neural network architecture
n = Network(
lossFunc = "Cross Entropy",
learningRate = 0.01,
optimisationFunc = "SGD", # Stochastic Gradient Descent
)
n.addLayer(3) # Input layer
n.addLayer(2, "ReLU") # Hidden layer
n.addLayer(1, "SoftMax") # Output layer
n.createWeightsAndBiases()
# Train the network
accuracy, averageLoss = n.train(mnist_images, mnist_labels, epochs = 10)
# Evaluate
print(f"Accuracy: {accuracy}%")
print(f"Average loss: {averageLoss}")
Note: This example assumes input and labels are preprocessed as NumPy arrays. You can use this script to download and save MNIST images using torchvision.
Benchmark
QuackNet vs PyTorch & TensorFlow
QuackNet was benchmarked against PyTorch and TensorFlow using the MNIST dataset,using identical architectures and hyperparameters to ensure fair comparison. Also each framework ran 5 times, and was averaged at the end.
Parameters:
- Neural Network Model Architecture: 784 (input) → 128 → 64 → 10 (output)
- Activation Function: Leaky ReLU for input and hidden layers, and SoftMax for output layer
- Optimiser: Gradient Descent with Batches
- Batch Size: 64
- Learning rate: 0.01
- Epochs 10
| Framework | Accuracy | Loss |
|---|---|---|
| QuackNet | 96.26% | 0.127 |
| PyTorch | 93.58% | 0.223 |
| TensorFlow | 94.88% | 0.175 |
Note: Due to differences in weight initialisation, dataset shuffling, and random seeds, QuackNet may perform slightly better some runs.
Benchmark scripts:
- The code for the QuackNet benchmark can be found here
- The code for the PyTorch benchmark can be found here
- The code for the TensorFlow benchmark can be found here
QuackNet benchmark on MNIST
The code for this benchmark can be found is the same as the one used to benchmark against PyTorch and TensorFlow.
Below is the graph showing the training accuracy and loss over 10 epochs, across 5 runs:
Learning Outcomes
ML Foundations
- Manual backpropagation (dense, convolutional, RNN BPTT, attention)
- Gradient derivation and chain rule
- Optimisers: SGD, GD, Adam, AdamW, RMSProp, Lion
Computer Science Practice
- 150+ unit tests, 91% coverage
- Modular, beginner friendly API design
- Efficient vectorised operations via NumPy
- Automated documentation with
pdoc
Roadmap
- Forward propagation Implemented the feed forward pass for neural network layers
- Activation functions Added support for Leaky ReLU, Sigmoid, SoftMax, and others
- Loss functions Implemented MSE, MAE, and Cross Entropy loss with their derivatives
- Backpropagation Completed backpropagation for gradient calculation and parameter updates
- Optimisers Added support for batching, stochastic gradient descent and gradient descent
- Convolutional Neural Network Implemented kernels, pooling and dense layers for Convolutional Neural Network
- Visualisation tools
Added support for visualising training, such as loss and accuracy graphs - Benchmark against PyTorch/TensorFlow Benchmark against popular machine learning frameworks on MNIST dataset
- Add Adam optimiser
Implement the Adam optimiser to improve training performance and convergence - Data augmentation Add data augmentation such as flipping, rotation and cropping
- Input Data augmentation: Add pixel normalisation of pixels and one-hot encoded label
- Skin Lesion detector
Use the neural network library to create a model for detecting skin lesions using HAM10000 for skin lesion images - Recurrent Neural Network Implement BPTT, with singular and stacked RNN implementation
- Transformer Implement Multi head Attention, residual connections, normalisation, positional encoding
Future Roadmap
QuackNet is now fully functional, stable and suitable for educational and experimental use. Future improvements are optional extensions to enchance capabilities and explore advanced research directions:
- Additional activation functions
Implement advanced activation functions (eg. GELU and Swish) - Additional regularisation Add L1/L2 regularisation and dropout to improve model generalisation and reduce overfitting
- Expanded demo projects Include additional example projects on more complex datasets to demonstrate practical applications of QuackNet
- Experimental optimisers Implement novel optimisation algorithms for experimentation and research purposes
Unit Tests
QuackNet includes unit tests that ensures the reliability of QuackNet's neural and convolutional components. They help to confirm that all layers and training processes behave as expected after every major update to ensure structural stability of all components. The tests are organised into directories:
- Unit Tests for NN - for standard neural network tests
- Unit Tests for CNN - for convolutional network specific tests
- Unit Tests for RNN - for recurrent network specific tests
- Unit Tests for Transformers - for transformer specific tests
- Unit Tests for Core code - for core code specific tests
These tests cover:
- Forward and backward propagation for both NN, CNN, RNN, and transformers
- Specific layers: Dense, Convolutional, Pooling, Multi head attention, Norm
- Activation/loss functions, including their derivatives
- Optimisation (SGD, GD, Adam, AdamW, RMSProp, Lion)
To run the tests:
pytest
To check test coverage:
coverage run -m pytest
coverage report -m
Related Projects
Skin Lesion Detector
A convolutional neural network (CNN) skin lesion classification model built with QuackNet, trained using the HAM10000 dataset. This model achieved 72% accuracy on a balanced validation set of skin lesion images.
You can explore the full project here: Skin Lesion Detector Repository
This project demonstrates how QuackNet can be applied to real-world image classification tasks.
Project Architecture
Neural Network Class
- Purpose Handles fully connected layers for standard neural network
- Key Components:
- Layers: Dense Layer
- Functions: Forward propagation, backpropagation
- Optimisers: SGD, GD, GD, Adam, AdamW, RMSProp, Lion
Convolutional Neural Network Class
- Purpose Specialised for image data processing using convolutional layers
- Key Components:
- Layers: Convolutional, pooling, dense and activation layers
- Functions: Forward propagation, backpropagation, flattening, global average pooling
- Optimisers: SGD, GD, GD, Adam, AdamW, RMSProp, Lion
RNN Class
- Purpose Handles sequential data with temporal dependencies
- Key Components:
- Layers: Singular RNN, Stacked RNN
- Functions: Forward propagation across time steps, backpropagation through time (BPTT)
- Optimisers: SGD, GD, GD, Adam, AdamW, RMSProp, Lion
Transformer Class
- Purpose Handles sequences with long range dependencies using attention mechanisms
- Key Components:
- Layers: Multi head self attention, position wise feed forward network, embeddings, positional encoding
- Functions: Forward propagation, residual connections, layer normalisation, backpropagation
- Optimisers: SGD, GD, GD, Adam, AdamW, RMSProp, Lion
License
This project is licensed under the MIT License - see the LICENSE file for details.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file quacknet-1.7.tar.gz.
File metadata
- Download URL: quacknet-1.7.tar.gz
- Upload date:
- Size: 39.5 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.13.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
081c6fcbf435bb949a49f58f09f5180492612c8ac572aa58887909617d240b75
|
|
| MD5 |
c76eaec00ecdf88e57b0d78cca89db7f
|
|
| BLAKE2b-256 |
07698a141cd96f096bf9b7d6771782e75b28c174d55bdbbe86146ffe4e2a1ecd
|
File details
Details for the file quacknet-1.7-py3-none-any.whl.
File metadata
- Download URL: quacknet-1.7-py3-none-any.whl
- Upload date:
- Size: 59.9 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.13.1
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
01449ec3217a5ec63f1ffb916fa81aceac4e03331bd088f6b3e846009e58f598
|
|
| MD5 |
b953c0b3e10bfa02b496fc8f3d94e8a9
|
|
| BLAKE2b-256 |
f04f8f521cc260d2eb7b994806a03f9214a97b0c843c20bd85e50e97a0d66ad2
|