diff --git a/README.md b/README.md index ff5ead3d..872935a4 100644 --- a/README.md +++ b/README.md @@ -1,15 +1,14 @@ -# Trace: End-to-end Generative Optimization for AI Systems and Agents +

+ drawing +

+# Trace: End-to-end Generative Optimization for AI Systems and Agents ![PyPI](https://img.shields.io/pypi/v/trace-opt) ![PyPI - Python Version](https://img.shields.io/pypi/pyversions/trace-opt) ![GitHub license](https://img.shields.io/badge/License-MIT-blue.svg) [![arXiv](https://img.shields.io/badge/arXiv-1234.56789-b31b1b.svg)](https://arxiv.org/abs/2406.16218) -

- drawing -

- Trace is a new AutoDiff-like tool for training AI systems end-to-end with general feedback (like numerical rewards or losses, natural language text, compiler errors, etc.). Trace generalizes the back-propagation algorithm by capturing and propagating an AI system's execution trace. Trace is implemented as a PyTorch-like Python library. Users write Python code directly and can use Trace primitives to optimize certain parts, just like training neural networks! diff --git a/docs/images/Trace_Black_B.png b/docs/images/Trace_Black_B.png new file mode 100644 index 00000000..52228dbd Binary files /dev/null and b/docs/images/Trace_Black_B.png differ