Metadata-Version: 2.4
Name: picodl-nn
Version: 0.1.0
Summary: A tiny deep learning library built from scratch requiring just numpy.
Requires-Python: >=3.12
Description-Content-Type: text/markdown
Requires-Dist: numpy
Dynamic: description
Dynamic: description-content-type
Dynamic: requires-dist
Dynamic: requires-python
Dynamic: summary


# picodl.

A tiny deep learning library built from scratch in Python - autograd engine, layers, losses, optimizers, and a training loop, all on top of plain numpy. No PyTorch, no TensorFlow, no hidden framework underneath.


## Documentation

[Documentation](https://picodl.vercel.app/docs.html)


## Installation

Install picodl with pip

```bash
  pip install picodl-nn
```
    
## Demo

A sample model

```python
from picodl.nn import NeuralNet
from picodl.layers import Linear, GELU
from picodl.loss import CrossEntropyLoss
from picodl.optim import AdamW
from picodl.train import train

net = NeuralNet([
    Linear(784, 128),
    GELU(),
    Linear(128, 10),
])

train(net, x_train, y_train,
      num_epochs=10,
      loss=CrossEntropyLoss(),
      optimizer=AdamW(lr=0.001, weight_decay=0.01))

net.save("model.picodl")
```
## Authors

- [@alight659](https://www.github.com/alight659)


## License

[MIT](https://github.com/alight659/picodl/blob/main/LICENSE)

