Skip to main content

Prompt. Generate Synthetic Data. Train & Align Models.

Project description

DataDreamer
https://datadreamer.dev

Prompt. Generate Synthetic Data. Train & Align Models.

Tests & Release Ruff

DataDreamer is a powerful open-source Python library for prompting, synthetic data generation, and training workflows. It is designed to be simple, extremely efficient, and research-grade.

Installation

pip3 install datadreamer.dev
demo.py Result of demo.py
                                                                                                   
demo.py

See the full demo script


                                                                                     
Demo

See the synthetic dataset and the trained model

🚀 For more demonstrations and recipes see the Quick Tour page.

With DataDreamer you can:

  • 💬 Create Prompting Workflows: Create and run multi-step, complex, prompting workflows easily with major open source or API-based LLMs.
  • 📊 Generate Synthetic Datasets: Generate synthetic datasets for novel tasks or augment existing datasets with LLMs.
  • ⚙️ Train Models: Align models. Fine-tune models. Instruction-tune models. Distill models. Train on existing data or synthetic data.
  • ... learn more about what's possible in the Overview Guide

DataDreamer is:

  • 🧩 Simple: Simple and approachable to use with sensible defaults, yet powerful with support for bleeding edge techniques.
  • 🔬 Research-Grade: Built for researchers, by researchers, but accessible to all. A focus on correctness, best practices, and reproducibility.
  • 🏎️ Efficient: Aggressive caching and resumability built-in. Support for techniques like quantization, parameter-efficient training (LoRA), and more.
  • 🔄 Reproducible: Workflows built with DataDreamer are easily shareable, reproducible, and extendable.
  • 🤝 Makes Sharing Easy: Publishing datasets and models is simple. Automatically generate data cards and model cards with metadata. Generate a list of any citations required.
  • ... learn more about the motivation and design principles behind DataDreamer.

Citation

coming soon...

Contact

Please reach out to us via email (ajayp@upenn.edu) or on Discord if you have any questions, comments, or feedback.



Copyright © 2024, Ajay Patel. Released under the MIT License.

Thank you to the maintainers at Hugging Face and LiteLLM for accepting contributions neccessary for DataDreamer and providing upstream support.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

datadreamer_dev-0.7.0.tar.gz (271.3 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

datadreamer_dev-0.7.0-py3-none-any.whl (360.8 kB view details)

Uploaded Python 3

File details

Details for the file datadreamer_dev-0.7.0.tar.gz.

File metadata

  • Download URL: datadreamer_dev-0.7.0.tar.gz
  • Upload date:
  • Size: 271.3 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.7.1 CPython/3.10.9 Linux/6.2.0-1017-aws

File hashes

Hashes for datadreamer_dev-0.7.0.tar.gz
Algorithm Hash digest
SHA256 245dad3b4230783e5face585ab424af0c7d860fba335c181f619127731f530ce
MD5 dad643ca343da627050a48d99e1cc1e9
BLAKE2b-256 b225aa701eb64f3022e64237dfa0b8bab8edf3c790ab616c06c9c3790d2fb4a3

See more details on using hashes here.

File details

Details for the file datadreamer_dev-0.7.0-py3-none-any.whl.

File metadata

  • Download URL: datadreamer_dev-0.7.0-py3-none-any.whl
  • Upload date:
  • Size: 360.8 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.7.1 CPython/3.10.9 Linux/6.2.0-1017-aws

File hashes

Hashes for datadreamer_dev-0.7.0-py3-none-any.whl
Algorithm Hash digest
SHA256 b3edcf72a5fe6b4f746b2214b2234396bb5ca8c36b20b631bdf4535a0fbcc662
MD5 47b3a50a22b8bafa69a15b5e43372500
BLAKE2b-256 3b89a4256112df186a559c1ba453801a8530dbdd812f8d7dfd7bcc45d20b7831

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page