Metadata-Version: 2.4
Name: embodied-ai-architect
Version: 0.9.0
Summary: Hardware/Software Codesign Space Exploration and Optimization Environment for Embodied AI systems
Author: Branes-ai Team
License: MIT
Requires-Python: >=3.11
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: torch>=2.0.0
Requires-Dist: numpy>=1.24.0
Requires-Dist: pydantic>=2.0.0
Requires-Dist: matplotlib>=3.7.0
Requires-Dist: jinja2>=3.1.0
Requires-Dist: click>=8.1.0
Requires-Dist: rich>=13.0.0
Provides-Extra: dev
Requires-Dist: pytest>=7.0.0; extra == "dev"
Requires-Dist: pytest-cov>=4.0.0; extra == "dev"
Requires-Dist: black>=23.0.0; extra == "dev"
Requires-Dist: ruff>=0.1.0; extra == "dev"
Requires-Dist: build>=1.0.0; extra == "dev"
Provides-Extra: remote
Requires-Dist: paramiko>=3.0.0; extra == "remote"
Provides-Extra: kubernetes
Requires-Dist: kubernetes>=28.0.0; extra == "kubernetes"
Provides-Extra: chat
Requires-Dist: anthropic>=0.39.0; extra == "chat"
Provides-Extra: optimization
Requires-Dist: pymoo>=0.6.0; extra == "optimization"
Requires-Dist: botorch>=0.11.0; extra == "optimization"
Requires-Dist: gpytorch>=1.12; extra == "optimization"
Requires-Dist: scipy>=1.10.0; extra == "optimization"
Provides-Extra: mcp
Requires-Dist: mcp>=1.0.0; extra == "mcp"
Requires-Dist: uvicorn>=0.25.0; extra == "mcp"
Requires-Dist: httpx>=0.25.0; extra == "mcp"
Provides-Extra: langgraph
Requires-Dist: langgraph>=0.2.0; extra == "langgraph"
Requires-Dist: langchain-anthropic>=0.3.0; extra == "langgraph"
Requires-Dist: langgraph-checkpoint-sqlite>=2.0.0; extra == "langgraph"
Provides-Extra: jetson
Requires-Dist: tensorrt>=8.6.0; extra == "jetson"
Requires-Dist: pycuda>=2022.1; extra == "jetson"
Requires-Dist: onnxruntime>=1.16.0; extra == "jetson"
Requires-Dist: Pillow>=9.0.0; extra == "jetson"
Provides-Extra: openvino
Requires-Dist: openvino>=2024.0.0; extra == "openvino"
Requires-Dist: nncf>=2.7.0; extra == "openvino"
Requires-Dist: onnxruntime>=1.16.0; extra == "openvino"
Requires-Dist: Pillow>=9.0.0; extra == "openvino"
Provides-Extra: coral
Requires-Dist: tensorflow>=2.13.0; extra == "coral"
Requires-Dist: onnx2tf>=1.20.0; extra == "coral"
Requires-Dist: onnxruntime>=1.16.0; extra == "coral"
Requires-Dist: Pillow>=9.0.0; extra == "coral"
Provides-Extra: all
Requires-Dist: paramiko>=3.0.0; extra == "all"
Requires-Dist: kubernetes>=28.0.0; extra == "all"
Requires-Dist: anthropic>=0.39.0; extra == "all"
Requires-Dist: langgraph>=0.2.0; extra == "all"
Requires-Dist: langchain-anthropic>=0.3.0; extra == "all"
Requires-Dist: onnxruntime>=1.16.0; extra == "all"
Requires-Dist: Pillow>=9.0.0; extra == "all"
Dynamic: license-file

# Embodied AI Architect

A design environment for creating and evaluating autonomous agents, with hardware/software codesign space exploration and optimization.

## Features

- **Model Analysis**: Analyze PyTorch model structure and compute requirements
- **Hardware Profiling**: Recommendations for edge/cloud deployment
- **Multi-Hardware Benchmarking**: Local CPU, remote SSH, Kubernetes backends
- **Interactive Chat**: Claude-powered architect for design decisions
- **Codebase Analysis**: Scan and assess application codebases for hardware deployment
- **SoC Optimization**: LangGraph-based RTL optimization loop (experimental)

## Installation

```bash
pip install embodied-ai-architect
```

With optional dependencies:

```bash
# Remote SSH benchmarking
pip install embodied-ai-architect[remote]

# Kubernetes benchmarking
pip install embodied-ai-architect[kubernetes]

# Interactive chat (requires ANTHROPIC_API_KEY)
pip install embodied-ai-architect[chat]

# All optional dependencies
pip install embodied-ai-architect[all]
```

## Usage

```bash
# Show available commands
branes --help

# Analyze a PyTorch model
branes analyze model.pt

# Run full workflow
branes workflow run model.pt

# Benchmark on local CPU
branes benchmark model.pt --backend local

# Scan and assess a codebase for hardware deployment
branes codebase scan /path/to/project
branes codebase assess /path/to/project --hardware jetson_orin

# Interactive chat session
export ANTHROPIC_API_KEY=your-key-here
branes chat
```

## Environment Variables

| Variable | Description |
|----------|-------------|
| `ANTHROPIC_API_KEY` | Required for Claude-powered features (chat, codebase analysis) |

## Documentation

For full documentation, development setup, and contributing guidelines, visit the
[GitHub repository](https://github.com/branes-ai/embodied-ai-architect).

## Related Projects

- [embodied-schemas](https://github.com/branes-ai/embodied-schemas): Shared Pydantic schemas and hardware catalog
- [graphs](https://github.com/branes-ai/graphs): Analysis tools and roofline models

## License

MIT License
