pytorch-kito 0.2.9__tar.gz

pytorch_kito-0.2.9/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Giuseppe Costantino
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

pytorch_kito-0.2.9/PKG-INFO

@@ -0,0 +1,332 @@
+Metadata-Version: 2.4
+Name: pytorch-kito
+Version: 0.2.9
+Summary: Effortless PyTorch training - define your model, Kito handles the rest
+Author-email: Giuseppe Costantino <giuseppe.costantino95@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2026 Giuseppe Costantino
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+Project-URL: Homepage, https://github.com/gcostantino/kito
+Project-URL: Documentation, https://kito.readthedocs.io
+Project-URL: Repository, https://github.com/gcostantino/kito
+Project-URL: Bug Tracker, https://github.com/gcostantino/kito/issues
+Keywords: pytorch,deep-learning,machine-learning,training,automation
+Classifier: Development Status :: 3 - Alpha
+Classifier: Intended Audience :: Science/Research
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.9
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Topic :: Scientific/Engineering
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.0.0
+Requires-Dist: numpy>=1.20.0
+Requires-Dist: h5py>=3.0.0
+Requires-Dist: torchsummary>=1.5.1
+Requires-Dist: packaging>=20.0
+Requires-Dist: tensorboard>=2.10.0
+Requires-Dist: pkbar>=0.5
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.0; extra == "dev"
+Requires-Dist: pytest-mock>=3.10; extra == "dev"
+Requires-Dist: black>=23.0; extra == "dev"
+Requires-Dist: ruff>=0.1.0; extra == "dev"
+Provides-Extra: tensorboard
+Requires-Dist: tensorboard>=2.10.0; extra == "tensorboard"
+Provides-Extra: docs
+Requires-Dist: sphinx>=5.0; extra == "docs"
+Requires-Dist: sphinx-rtd-theme>=1.0; extra == "docs"
+Provides-Extra: all
+Requires-Dist: tensorboard>=2.10.0; extra == "all"
+Requires-Dist: pytest>=7.0; extra == "all"
+Requires-Dist: pytest-cov>=4.0; extra == "all"
+Requires-Dist: pytest-mock>=3.10; extra == "all"
+Requires-Dist: black>=23.0; extra == "all"
+Requires-Dist: ruff>=0.1.0; extra == "all"
+Requires-Dist: sphinx>=5.0; extra == "all"
+Requires-Dist: sphinx-rtd-theme>=1.0; extra == "all"
+Dynamic: license-file
+
+<p align="center">
+  <img src="https://raw.githubusercontent.com/gcostantino/kito/refs/heads/main/assets/kito_simplelogo.png" alt="Kito Logo" width="200"/>
+</p>
+
+---
+**Effortless PyTorch training - define your model, Kito handles the rest.**
+
+[![Tests](https://github.com/gcostantino/kito/actions/workflows/tests.yml/badge.svg)](https://github.com/gcostantino/kito/actions)
+[![PyPI version](https://img.shields.io/pypi/v/pytorch-kito)](https://pypi.org/project/pytorch-kito/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/pytorch-kito)](https://pypi.org/project/pytorch-kito/)
+[![Downloads](https://img.shields.io/pypi/dm/pytorch-kito?style=round-square)](https://pepy.tech/project/pytorch-kito)
+[![Documentation Status](https://readthedocs.org/projects/kito/badge/?version=latest)](https://kito.readthedocs.io/en/latest/?badge=latest)
+
+Kito is a lightweight PyTorch training library that eliminates boilerplate code. Define your model architecture and loss function - Kito automatically handles training loops, optimization, callbacks, distributed training, and more.
+
+## ✨ Key Features
+
+- **Zero Boilerplate** - No training loops, no optimizer setup, no device management
+- **Auto-Everything** - Automatic model building, optimizer binding, and device assignment
+- **Built-in DDP** - Distributed training works out of the box
+- **Smart Callbacks** - TensorBoard, checkpointing, logging, and custom callbacks
+- **Flexible** - Simple for beginners, powerful for experts
+- **Lightweight** - Minimal dependencies, pure PyTorch under the hood
+
+## Quick Start
+
+### Installation
+
+```bash
+pip install pytorch-kito
+```
+
+### Your First Model in 3 Steps
+
+```python
+import torch.nn as nn
+from kito import Engine, KitoModule
+
+# 1. Define your model
+class MyModel(KitoModule):
+    def build_inner_model(self):
+        self.model = nn.Sequential(
+            nn.Linear(784, 128),
+            nn.ReLU(),
+            nn.Linear(128, 10)
+        )
+        self.model_input_size = (784,)
+
+    def bind_optimizer(self):
+        self.optimizer = torch.optim.Adam(
+            self.model.parameters(),
+            lr=self.learning_rate
+        )
+
+# 2. Initialize
+model = MyModel('MyModel', device, config)
+engine = Engine(model, config)
+
+# 3. Train! (That's it - everything else is automatic)
+engine.fit(train_loader, val_loader, max_epochs=10)
+```
+
+## Philosophy
+
+Kito follows a **"define once, train anywhere"** philosophy:
+
+1. **You focus on**: Model architecture and research ideas
+2. **Kito handles**: Training loops, optimization, distributed training, callbacks
+
+Perfect for researchers who want to iterate quickly without rewriting training code.
+
+## Core Concepts
+
+### KitoModule
+
+Your model inherits from `KitoModule` and implements two methods:
+
+```python
+class MyModel(KitoModule):
+    def build_inner_model(self):
+        # Define your architecture
+        self.model = nn.Sequential(...)
+        self.model_input_size = (C, H, W)  # Input shape
+
+    def bind_optimizer(self):
+        # Choose your optimizer
+        self.optimizer = torch.optim.Adam(
+            self.model.parameters(),
+            lr=self.learning_rate
+        )
+```
+
+### Engine
+
+The `Engine` orchestrates everything:
+
+```python
+engine = Engine(module, config)
+
+# Training
+engine.fit(train_loader, val_loader, max_epochs=100)
+
+# Inference
+predictions = engine.predict(test_loader)
+```
+
+### Data Pipeline
+
+Kito provides a clean data pipeline with preprocessing:
+
+```python
+from kito.data import H5Dataset, GenericDataPipeline
+from kito.data.preprocessing import Pipeline, Normalize, ToTensor
+
+# Create dataset
+dataset = H5Dataset('data.h5')
+
+# Add preprocessing
+preprocessing = Pipeline([
+    Normalize(min_val=0.0, max_val=1.0),
+    ToTensor()
+])
+
+# Setup data pipeline
+pipeline = GenericDataPipeline(
+    config=config,
+    dataset=dataset,
+    preprocessing=preprocessing
+)
+pipeline.setup()
+
+# Get dataloaders
+train_loader = pipeline.train_dataloader()
+val_loader = pipeline.val_dataloader()
+```
+
+### Callbacks
+
+Kito includes powerful callbacks for common tasks:
+
+```python
+from kito.callbacks import ModelCheckpoint, EarlyStopping, CSVLogger
+
+callbacks = [
+    ModelCheckpoint('best_model.pt', monitor='val_loss', mode='min'),
+    EarlyStopping(patience=10, monitor='val_loss'),
+    CSVLogger('training.csv')
+]
+
+engine.fit(train_loader, val_loader, callbacks=callbacks)
+```
+
+Or create custom callbacks:
+
+```python
+from kito.callbacks import Callback
+
+class MyCallback(Callback):
+    def on_epoch_end(self, epoch, logs, **kwargs):
+        print(f"Epoch {epoch}: loss={logs['train_loss']:.4f}")
+```
+
+## Advanced Features
+
+### Distributed Training (DDP)
+
+Enable distributed training with one config change:
+
+```python
+config.training.distributed_training = True
+
+# Everything else stays the same!
+engine.fit(train_loader, val_loader, max_epochs=100)
+```
+
+### Custom Training Logic
+
+Override `training_step` for custom behavior:
+
+```python
+class MyModel(KitoModule):
+    def training_step(self, batch, pbar_handler=None):
+        inputs, targets = batch
+
+        # Custom forward pass
+        outputs = self.model(inputs)
+        loss = self.compute_loss((inputs, targets), outputs)
+
+        # Custom backward (e.g., gradient clipping)
+        self.optimizer.zero_grad()
+        loss.backward()
+        torch.nn.utils.clip_grad_norm_(self.model.parameters(), max_norm=1.0)
+        self.optimizer.step()
+
+        return {'loss': loss}
+```
+
+### Multiple Datasets
+
+Kito supports HDF5 and in-memory datasets out of the box:
+
+```python
+from kito.data import H5Dataset, MemDataset
+
+# HDF5 dataset (lazy loading)
+dataset = H5Dataset('large_data.h5')
+
+# In-memory dataset (fast)
+dataset = MemDataset(x_train, y_train)
+```
+
+Register custom datasets easily:
+
+```python
+from kito.data.registry import DATASETS
+
+@DATASETS.register('my_custom_dataset')
+class MyDataset(KitoDataset):
+    def _load_sample(self, index):
+        return data, labels
+```
+
+## 📦 Installation Options
+
+```bash
+# Basic installation
+pip install pytorch-kito
+
+# With TensorBoard support
+pip install pytorch-kito[tensorboard]
+
+# Development installation
+pip install pytorch-kito[dev]
+
+# Everything
+pip install pytorch-kito[all]
+```
+
+## 🤝 Contributing
+
+Contributions are very welcome! Please check out our [Contributing Guide](CONTRIBUTING.md).
+
+## 📄 License
+
+MIT License - see [LICENSE](LICENSE) file for details.
+
+## 🙏 Acknowledgments
+
+Kito is inspired by PyTorch Lightning and Keras, aiming to bring similar ease-of-use to pure PyTorch workflows for researchers.
+
+## Contact
+
+- **GitHub Issues**: [Report bugs or request features](https://github.com/gcostantino/kito/issues)
+- **GitHub Discussions**: [Ask questions](https://github.com/gcostantino/kito/discussions)
+
+---
+
+**Made with ❤️ for the PyTorch community**

pytorch_kito-0.2.9/README.md

@@ -0,0 +1,259 @@
+<p align="center">
+  <img src="https://raw.githubusercontent.com/gcostantino/kito/refs/heads/main/assets/kito_simplelogo.png" alt="Kito Logo" width="200"/>
+</p>
+
+---
+**Effortless PyTorch training - define your model, Kito handles the rest.**
+
+[![Tests](https://github.com/gcostantino/kito/actions/workflows/tests.yml/badge.svg)](https://github.com/gcostantino/kito/actions)
+[![PyPI version](https://img.shields.io/pypi/v/pytorch-kito)](https://pypi.org/project/pytorch-kito/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+[![PyPI - Python Version](https://img.shields.io/pypi/pyversions/pytorch-kito)](https://pypi.org/project/pytorch-kito/)
+[![Downloads](https://img.shields.io/pypi/dm/pytorch-kito?style=round-square)](https://pepy.tech/project/pytorch-kito)
+[![Documentation Status](https://readthedocs.org/projects/kito/badge/?version=latest)](https://kito.readthedocs.io/en/latest/?badge=latest)
+
+Kito is a lightweight PyTorch training library that eliminates boilerplate code. Define your model architecture and loss function - Kito automatically handles training loops, optimization, callbacks, distributed training, and more.
+
+## ✨ Key Features
+
+- **Zero Boilerplate** - No training loops, no optimizer setup, no device management
+- **Auto-Everything** - Automatic model building, optimizer binding, and device assignment
+- **Built-in DDP** - Distributed training works out of the box
+- **Smart Callbacks** - TensorBoard, checkpointing, logging, and custom callbacks
+- **Flexible** - Simple for beginners, powerful for experts
+- **Lightweight** - Minimal dependencies, pure PyTorch under the hood
+
+## Quick Start
+
+### Installation
+
+```bash
+pip install pytorch-kito
+```
+
+### Your First Model in 3 Steps
+
+```python
+import torch.nn as nn
+from kito import Engine, KitoModule
+
+# 1. Define your model
+class MyModel(KitoModule):
+    def build_inner_model(self):
+        self.model = nn.Sequential(
+            nn.Linear(784, 128),
+            nn.ReLU(),
+            nn.Linear(128, 10)
+        )
+        self.model_input_size = (784,)
+
+    def bind_optimizer(self):
+        self.optimizer = torch.optim.Adam(
+            self.model.parameters(),
+            lr=self.learning_rate
+        )
+
+# 2. Initialize
+model = MyModel('MyModel', device, config)
+engine = Engine(model, config)
+
+# 3. Train! (That's it - everything else is automatic)
+engine.fit(train_loader, val_loader, max_epochs=10)
+```
+
+## Philosophy
+
+Kito follows a **"define once, train anywhere"** philosophy:
+
+1. **You focus on**: Model architecture and research ideas
+2. **Kito handles**: Training loops, optimization, distributed training, callbacks
+
+Perfect for researchers who want to iterate quickly without rewriting training code.
+
+## Core Concepts
+
+### KitoModule
+
+Your model inherits from `KitoModule` and implements two methods:
+
+```python
+class MyModel(KitoModule):
+    def build_inner_model(self):
+        # Define your architecture
+        self.model = nn.Sequential(...)
+        self.model_input_size = (C, H, W)  # Input shape
+
+    def bind_optimizer(self):
+        # Choose your optimizer
+        self.optimizer = torch.optim.Adam(
+            self.model.parameters(),
+            lr=self.learning_rate
+        )
+```
+
+### Engine
+
+The `Engine` orchestrates everything:
+
+```python
+engine = Engine(module, config)
+
+# Training
+engine.fit(train_loader, val_loader, max_epochs=100)
+
+# Inference
+predictions = engine.predict(test_loader)
+```
+
+### Data Pipeline
+
+Kito provides a clean data pipeline with preprocessing:
+
+```python
+from kito.data import H5Dataset, GenericDataPipeline
+from kito.data.preprocessing import Pipeline, Normalize, ToTensor
+
+# Create dataset
+dataset = H5Dataset('data.h5')
+
+# Add preprocessing
+preprocessing = Pipeline([
+    Normalize(min_val=0.0, max_val=1.0),
+    ToTensor()
+])
+
+# Setup data pipeline
+pipeline = GenericDataPipeline(
+    config=config,
+    dataset=dataset,
+    preprocessing=preprocessing
+)
+pipeline.setup()
+
+# Get dataloaders
+train_loader = pipeline.train_dataloader()
+val_loader = pipeline.val_dataloader()
+```
+
+### Callbacks
+
+Kito includes powerful callbacks for common tasks:
+
+```python
+from kito.callbacks import ModelCheckpoint, EarlyStopping, CSVLogger
+
+callbacks = [
+    ModelCheckpoint('best_model.pt', monitor='val_loss', mode='min'),
+    EarlyStopping(patience=10, monitor='val_loss'),
+    CSVLogger('training.csv')
+]
+
+engine.fit(train_loader, val_loader, callbacks=callbacks)
+```
+
+Or create custom callbacks:
+
+```python
+from kito.callbacks import Callback
+
+class MyCallback(Callback):
+    def on_epoch_end(self, epoch, logs, **kwargs):
+        print(f"Epoch {epoch}: loss={logs['train_loss']:.4f}")
+```
+
+## Advanced Features
+
+### Distributed Training (DDP)
+
+Enable distributed training with one config change:
+
+```python
+config.training.distributed_training = True
+
+# Everything else stays the same!
+engine.fit(train_loader, val_loader, max_epochs=100)
+```
+
+### Custom Training Logic
+
+Override `training_step` for custom behavior:
+
+```python
+class MyModel(KitoModule):
+    def training_step(self, batch, pbar_handler=None):
+        inputs, targets = batch
+
+        # Custom forward pass
+        outputs = self.model(inputs)
+        loss = self.compute_loss((inputs, targets), outputs)
+
+        # Custom backward (e.g., gradient clipping)
+        self.optimizer.zero_grad()
+        loss.backward()
+        torch.nn.utils.clip_grad_norm_(self.model.parameters(), max_norm=1.0)
+        self.optimizer.step()
+
+        return {'loss': loss}
+```
+
+### Multiple Datasets
+
+Kito supports HDF5 and in-memory datasets out of the box:
+
+```python
+from kito.data import H5Dataset, MemDataset
+
+# HDF5 dataset (lazy loading)
+dataset = H5Dataset('large_data.h5')
+
+# In-memory dataset (fast)
+dataset = MemDataset(x_train, y_train)
+```
+
+Register custom datasets easily:
+
+```python
+from kito.data.registry import DATASETS
+
+@DATASETS.register('my_custom_dataset')
+class MyDataset(KitoDataset):
+    def _load_sample(self, index):
+        return data, labels
+```
+
+## 📦 Installation Options
+
+```bash
+# Basic installation
+pip install pytorch-kito
+
+# With TensorBoard support
+pip install pytorch-kito[tensorboard]
+
+# Development installation
+pip install pytorch-kito[dev]
+
+# Everything
+pip install pytorch-kito[all]
+```
+
+## 🤝 Contributing
+
+Contributions are very welcome! Please check out our [Contributing Guide](CONTRIBUTING.md).
+
+## 📄 License
+
+MIT License - see [LICENSE](LICENSE) file for details.
+
+## 🙏 Acknowledgments
+
+Kito is inspired by PyTorch Lightning and Keras, aiming to bring similar ease-of-use to pure PyTorch workflows for researchers.
+
+## Contact
+
+- **GitHub Issues**: [Report bugs or request features](https://github.com/gcostantino/kito/issues)
+- **GitHub Discussions**: [Ask questions](https://github.com/gcostantino/kito/discussions)
+
+---
+
+**Made with ❤️ for the PyTorch community**

pytorch_kito-0.2.9/pyproject.toml

@@ -0,0 +1,89 @@
+[build-system]
+requires = ["setuptools>=61.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "pytorch-kito"
+dynamic = ["version"]
+description = "Effortless PyTorch training - define your model, Kito handles the rest"
+readme = "README.md"
+requires-python = ">=3.9"
+license = {file = "LICENSE"}
+authors = [
+    {name = "Giuseppe Costantino", email = "giuseppe.costantino95@gmail.com"}
+]
+keywords = ["pytorch", "deep-learning", "machine-learning", "training", "automation"]
+
+#   3 - Alpha
+  #   4 - Beta
+  #   5 - Production/Stable
+
+classifiers = [
+    "Development Status :: 3 - Alpha",
+    "Intended Audience :: Science/Research",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Topic :: Scientific/Engineering",
+    "Topic :: Scientific/Engineering :: Artificial Intelligence",
+]
+
+# Core dependencies (required for kito to work)
+dependencies = [
+    "torch>=2.0.0",
+    "numpy>=1.20.0",
+    "h5py>=3.0.0",
+    "torchsummary>=1.5.1",
+    "packaging>=20.0",
+    "tensorboard>=2.10.0",
+    "pkbar>=0.5",
+]
+
+# Optional dependencies
+[project.optional-dependencies]
+# Development and testing
+dev = [
+    "pytest>=7.0",
+    "pytest-cov>=4.0",
+    "pytest-mock>=3.10",
+    "black>=23.0",
+    "ruff>=0.1.0",
+]
+
+# TensorBoard support
+tensorboard = [
+    "tensorboard>=2.10.0",
+]
+
+# Documentation
+docs = [
+    "sphinx>=5.0",
+    "sphinx-rtd-theme>=1.0",
+]
+
+# All extras (convenience)
+all = [
+    "tensorboard>=2.10.0",
+    "pytest>=7.0",
+    "pytest-cov>=4.0",
+    "pytest-mock>=3.10",
+    "black>=23.0",
+    "ruff>=0.1.0",
+    "sphinx>=5.0",
+    "sphinx-rtd-theme>=1.0",
+]
+
+[project.urls]
+Homepage = "https://github.com/gcostantino/kito"
+Documentation = "https://kito.readthedocs.io"
+Repository = "https://github.com/gcostantino/kito"
+"Bug Tracker" = "https://github.com/gcostantino/kito/issues"
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.dynamic]
+version = {attr = "kito.__version__"}

pytorch_kito-0.2.9/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+