frameworm 1.0.0__tar.gz

frameworm-1.0.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Aakash Ali
+
+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.

frameworm-1.0.0/PKG-INFO

@@ -0,0 +1,160 @@
+Metadata-Version: 2.4
+Name: frameworm
+Version: 1.0.0
+Summary: Complete Machine Learning Framework for Production
+Author-email: Aakash Ali <Aakashali0440@gmail.com>
+License: MIT License
+        
+        Copyright (c) 2026 Aakash Ali
+        
+        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/Aakash0440/frameworm
+Project-URL: Documentation, https://frameworm.readthedocs.io
+Project-URL: Repository, https://github.com/Aakash0440/frameworm
+Project-URL: Bug Tracker, https://github.com/Aakash0440/frameworm/issues
+Project-URL: Changelog, https://github.com/Aakash0440/frameworm/blob/main/CHANGELOG.md
+Keywords: machine-learning,deep-learning,pytorch,training,deployment,experiments
+Classifier: Development Status :: 5 - Production/Stable
+Classifier: Intended Audience :: Developers
+Classifier: Intended Audience :: Science/Research
+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: Programming Language :: Python :: 3.12
+Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
+Requires-Python: >=3.14
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torch>=2.0.0
+Requires-Dist: torchvision>=0.15.0
+Requires-Dist: numpy>=1.23.0
+Requires-Dist: pyyaml>=6.0
+Requires-Dist: click>=8.0.0
+Requires-Dist: click-completion>=0.5.0
+Requires-Dist: rich>=13.0.0
+Requires-Dist: tqdm>=4.65.0
+Provides-Extra: search
+Requires-Dist: scikit-optimize>=0.9.0; extra == "search"
+Provides-Extra: deployment
+Requires-Dist: fastapi>=0.104.0; extra == "deployment"
+Requires-Dist: uvicorn[standard]>=0.24.0; extra == "deployment"
+Requires-Dist: onnx>=1.15.0; extra == "deployment"
+Requires-Dist: onnxruntime>=1.16.0; extra == "deployment"
+Provides-Extra: dashboard
+Requires-Dist: fastapi>=0.104.0; extra == "dashboard"
+Requires-Dist: uvicorn[standard]>=0.24.0; extra == "dashboard"
+Requires-Dist: psutil>=5.9.0; extra == "dashboard"
+Provides-Extra: metrics
+Requires-Dist: lpips>=0.1.4; extra == "metrics"
+Provides-Extra: dev
+Requires-Dist: pytest>=7.4.0; extra == "dev"
+Requires-Dist: pytest-cov>=4.1.0; extra == "dev"
+Requires-Dist: black>=23.0.0; extra == "dev"
+Requires-Dist: flake8>=6.0.0; extra == "dev"
+Requires-Dist: mypy>=1.5.0; extra == "dev"
+Requires-Dist: isort>=5.12.0; extra == "dev"
+Requires-Dist: pre-commit>=3.4.0; extra == "dev"
+Provides-Extra: all
+Requires-Dist: frameworm[dashboard,deployment,metrics,search]; extra == "all"
+Dynamic: license-file
+
+# FRAMEWORM
+
+**Complete Machine Learning Framework for Production**
+
+[![PyPI](https://img.shields.io/pypi/v/frameworm)](https://pypi.org/project/frameworm/)
+[![Python](https://img.shields.io/pypi/pyversions/frameworm)](https://python.org)
+[![Tests](https://img.shields.io/github/workflow/status/yourusername/frameworm/tests)](https://github.com/yourusername/frameworm/actions)
+[![Coverage](https://img.shields.io/codecov/c/github/yourusername/frameworm)](https://codecov.io/gh/yourusername/frameworm)
+[![Documentation](https://img.shields.io/badge/docs-mkdocs-blue)](https://frameworm.readthedocs.io)
+[![License](https://img.shields.io/github/license/yourusername/frameworm)](https://github.com/yourusername/frameworm/blob/main/LICENSE)
+[![Discord](https://img.shields.io/discord/123456789)](https://discord.gg/frameworm)
+
+---
+
+## Features
+
+- 🚀 **Simple API** - Train models in minutes
+- 📊 **Experiment Tracking** - Built-in, no external servers
+- 🔍 **Hyperparameter Search** - Grid, random, Bayesian
+- 🎯 **Production Ready** - Export, serve, deploy
+- ⚡ **Fast** - Multi-GPU, distributed, mixed precision
+- 🎨 **Web Dashboard** - Beautiful UI for tracking
+
+---
+
+## Quick Start
+```bash
+pip install frameworm
+frameworm init my-project
+cd my-project
+frameworm train --config config.yaml
+```
+
+See [documentation](https://frameworm.readthedocs.io) for more.
+
+---
+
+## Why FRAMEWORM?
+
+| Feature | FRAMEWORM | PyTorch Lightning | Others |
+|---------|-----------|-------------------|--------|
+| Training | ✅ | ✅ | ✅ |
+| Experiment Tracking | ✅ Built-in | ⚠️ External | ⚠️ External |
+| Hyperparameter Search | ✅ Built-in | ❌ | ⚠️ Limited |
+| Deployment | ✅ Built-in | ❌ | ⚠️ Limited |
+| Web UI | ✅ Built-in | ❌ | ❌ |
+| Complexity | Low | Medium | High |
+
+---
+
+## Documentation
+
+- [Quick Start](https://frameworm.readthedocs.io/getting-started/quickstart/)
+- [User Guide](https://frameworm.readthedocs.io/user-guide/training/)
+- [Tutorials](https://frameworm.readthedocs.io/tutorials/vae-tutorial/)
+- [API Reference](https://frameworm.readthedocs.io/api-reference/core/)
+- [Examples](https://frameworm.readthedocs.io/examples/basic-training/)
+
+---
+
+## Contributing
+
+See [CONTRIBUTING.md](CONTRIBUTING.md)
+
+---
+
+## License
+
+MIT License - see [LICENSE](LICENSE)
+
+---
+
+## Citation
+```bibtex
+@software{frameworm2024,
+  title = {FRAMEWORM: Complete Machine Learning Framework},
+  author = {Aakash Ali},
+  year = {2026},
+  url = {https://github.com/Aakash0440/frameworm}
+}
+```

frameworm-1.0.0/README.md

@@ -0,0 +1,81 @@
+# FRAMEWORM
+
+**Complete Machine Learning Framework for Production**
+
+[![PyPI](https://img.shields.io/pypi/v/frameworm)](https://pypi.org/project/frameworm/)
+[![Python](https://img.shields.io/pypi/pyversions/frameworm)](https://python.org)
+[![Tests](https://img.shields.io/github/workflow/status/yourusername/frameworm/tests)](https://github.com/yourusername/frameworm/actions)
+[![Coverage](https://img.shields.io/codecov/c/github/yourusername/frameworm)](https://codecov.io/gh/yourusername/frameworm)
+[![Documentation](https://img.shields.io/badge/docs-mkdocs-blue)](https://frameworm.readthedocs.io)
+[![License](https://img.shields.io/github/license/yourusername/frameworm)](https://github.com/yourusername/frameworm/blob/main/LICENSE)
+[![Discord](https://img.shields.io/discord/123456789)](https://discord.gg/frameworm)
+
+---
+
+## Features
+
+- 🚀 **Simple API** - Train models in minutes
+- 📊 **Experiment Tracking** - Built-in, no external servers
+- 🔍 **Hyperparameter Search** - Grid, random, Bayesian
+- 🎯 **Production Ready** - Export, serve, deploy
+- ⚡ **Fast** - Multi-GPU, distributed, mixed precision
+- 🎨 **Web Dashboard** - Beautiful UI for tracking
+
+---
+
+## Quick Start
+```bash
+pip install frameworm
+frameworm init my-project
+cd my-project
+frameworm train --config config.yaml
+```
+
+See [documentation](https://frameworm.readthedocs.io) for more.
+
+---
+
+## Why FRAMEWORM?
+
+| Feature | FRAMEWORM | PyTorch Lightning | Others |
+|---------|-----------|-------------------|--------|
+| Training | ✅ | ✅ | ✅ |
+| Experiment Tracking | ✅ Built-in | ⚠️ External | ⚠️ External |
+| Hyperparameter Search | ✅ Built-in | ❌ | ⚠️ Limited |
+| Deployment | ✅ Built-in | ❌ | ⚠️ Limited |
+| Web UI | ✅ Built-in | ❌ | ❌ |
+| Complexity | Low | Medium | High |
+
+---
+
+## Documentation
+
+- [Quick Start](https://frameworm.readthedocs.io/getting-started/quickstart/)
+- [User Guide](https://frameworm.readthedocs.io/user-guide/training/)
+- [Tutorials](https://frameworm.readthedocs.io/tutorials/vae-tutorial/)
+- [API Reference](https://frameworm.readthedocs.io/api-reference/core/)
+- [Examples](https://frameworm.readthedocs.io/examples/basic-training/)
+
+---
+
+## Contributing
+
+See [CONTRIBUTING.md](CONTRIBUTING.md)
+
+---
+
+## License
+
+MIT License - see [LICENSE](LICENSE)
+
+---
+
+## Citation
+```bibtex
+@software{frameworm2024,
+  title = {FRAMEWORM: Complete Machine Learning Framework},
+  author = {Aakash Ali},
+  year = {2026},
+  url = {https://github.com/Aakash0440/frameworm}
+}
+```

frameworm-1.0.0/cli/__init__.py

@@ -0,0 +1,93 @@
+"""
+Project initialization.
+"""
+
+import shutil
+from pathlib import Path
+
+from click import echo
+
+
+def create_project(project_path: Path, template: str = "basic"):
+    """Create new FRAMEWORM project"""
+
+    # Create directory structure
+    project_path.mkdir(parents=True, exist_ok=True)
+
+    (project_path / "configs").mkdir(exist_ok=True)
+    (project_path / "data").mkdir(exist_ok=True)
+    (project_path / "experiments").mkdir(exist_ok=True)
+    (project_path / "checkpoints").mkdir(exist_ok=True)
+    (project_path / "scripts").mkdir(exist_ok=True)
+
+    # Create README
+    readme = project_path / "README.md"
+    readme.write_text(f"""# {project_path.name}
+
+FRAMEWORM project created with template: {template}
+
+## Quick Start
+```bash
+# Train model
+frameworm train --config configs/config.yaml
+
+# Evaluate
+frameworm evaluate --checkpoint checkpoints/best.pt
+
+# Export
+frameworm export checkpoints/best.pt --format onnx
+
+# Serve
+frameworm serve exported/model.pt
+```
+
+## Structure
+
+- `configs/` - Configuration files
+- `data/` - Datasets
+- `experiments/` - Experiment tracking
+- `checkpoints/` - Model checkpoints
+- `scripts/` - Custom scripts
+""")
+
+    # Create basic config
+    config = project_path / "configs" / "config.yaml"
+    config.write_text(f"""# FRAMEWORM Configuration
+
+model:
+  type: {template if template != 'basic' else 'vae'}
+  latent_dim: 128
+  hidden_dim: 256
+
+training:
+  epochs: 100
+  batch_size: 128
+  lr: 0.001
+  device: cuda
+
+data:
+  root: data/
+  image_size: 64
+""")
+
+    # Create .gitignore
+    gitignore = project_path / ".gitignore"
+    gitignore.write_text("""__pycache__/
+*.pyc
+*.pyo
+*.pyd
+.Python
+*.so
+*.egg
+*.egg-info/
+dist/
+build/
+data/
+experiments/
+checkpoints/
+.DS_Store
+""")
+
+    echo(f"✓ Created directory structure")
+    echo(f"✓ Created README.md")
+    echo(f"✓ Created config.yaml")

frameworm-1.0.0/cli/evaluate.py

@@ -0,0 +1,47 @@
+"""
+Model evaluation command.
+"""
+
+import torch
+from click import echo
+
+from core import Config, get_model
+from metrics import MetricEvaluator
+
+
+def run_evaluation(config_path: str, checkpoint_path: str, metrics: list, num_samples: int = 10000):
+    """Run model evaluation"""
+
+    # Load config
+    config = Config(config_path)
+
+    # Load checkpoint
+    checkpoint = torch.load(checkpoint_path)
+
+    # Create model
+    model = get_model(config.model.type)(config)
+    model.load_state_dict(checkpoint["model_state_dict"])
+    model.eval()
+
+    echo(f"✓ Model loaded from: {checkpoint_path}")
+
+    # Create evaluator
+    evaluator = MetricEvaluator(
+        metrics=metrics,
+        real_data=None,  # Placeholder
+        device="cuda" if torch.cuda.is_available() else "cpu",
+    )
+
+    echo(f"Computing metrics: {metrics}")
+
+    # Evaluate
+    results = evaluator.evaluate(model, num_samples=num_samples)
+
+    echo("\n" + "=" * 60)
+    echo("EVALUATION RESULTS")
+    echo("=" * 60)
+
+    for metric, value in results.items():
+        echo(f"{metric}: {value:.4f}")
+
+    echo("=" * 60)

frameworm-1.0.0/cli/export.py

@@ -0,0 +1,65 @@
+"""
+Model export command.
+"""
+
+from pathlib import Path
+
+import torch
+from click import echo
+
+from deployment import ModelExporter
+
+
+def export_model(
+    checkpoint_path: str,
+    export_format: str = "all",
+    output_dir: str = "exported",
+    quantize: bool = False,
+    benchmark: bool = False,
+):
+    """Export model to deployment formats"""
+
+    # Load checkpoint
+    checkpoint = torch.load(checkpoint_path)
+    model = checkpoint.get("model")  # Or reconstruct from state_dict
+
+    if model is None:
+        echo("✗ Model not found in checkpoint")
+        return
+
+    # Create output directory
+    output_path = Path(output_dir)
+    output_path.mkdir(exist_ok=True, parents=True)
+
+    # Example input (you'd get this from config)
+    example_input = torch.randn(1, 3, 64, 64)
+
+    # Create exporter
+    exporter = ModelExporter(model, example_input)
+
+    # Export formats
+    formats_to_export = []
+    if export_format == "all":
+        formats_to_export = ["torchscript", "onnx"]
+    else:
+        formats_to_export = [export_format]
+
+    for fmt in formats_to_export:
+        output_file = output_path / f"model.{fmt.replace('torchscript', 'pt')}"
+
+        if fmt == "torchscript":
+            exporter.to_torchscript(str(output_file))
+        elif fmt == "onnx":
+            exporter.to_onnx(str(output_file))
+
+    # Quantize if requested
+    if quantize:
+        quant_file = output_path / "model_quant.pt"
+        exporter.quantize(str(quant_file))
+
+    # Benchmark if requested
+    if benchmark:
+        echo("\nBenchmarking...")
+        exporter.benchmark_inference()
+
+    echo(f"\n✓ Export complete: {output_path}")

frameworm-1.0.0/cli/logger.py

@@ -0,0 +1,30 @@
+"""
+CLI logging utilities.
+"""
+
+import logging
+
+from rich.logging import RichHandler
+
+
+def setup_logging(level: str = "INFO", log_file: str = None):
+    """Setup CLI logging"""
+
+    # Create logger
+    logger = logging.getLogger("frameworm")
+    logger.setLevel(getattr(logging, level.upper()))
+
+    # Console handler with rich formatting
+    console_handler = RichHandler(rich_tracebacks=True)
+    console_handler.setFormatter(logging.Formatter("%(message)s"))
+    logger.addHandler(console_handler)
+
+    # File handler if specified
+    if log_file:
+        file_handler = logging.FileHandler(log_file)
+        file_handler.setFormatter(
+            logging.Formatter("%(asctime)s - %(name)s - %(levelname)s - %(message)s")
+        )
+        logger.addHandler(file_handler)
+
+    return logger