Refactor.

Added support for exporting models to onnx and logging them on mlflow.

Add logging of images of difficult examples.

Add context manager for logging with backoff.

Force alphabetical sort with ruff.

Add logging of histograms with model weights.

Added code for the CNN.

Import Callable from collections.abc.

Update logging of metrics.

Add some stubs for torchvision.transform

Add code for MLP of Chapter 2.

Add utils for mlflow tracking.

Update dependencies.

Add more utils.

Added more dependencies.