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.4.0.tar.gz (271.0 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.4.0-py3-none-any.whl (360.6 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: datadreamer_dev-0.4.0.tar.gz
  • Upload date:
  • Size: 271.0 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.4.0.tar.gz
Algorithm Hash digest
SHA256 1bed0a49befcc8e24193c8b732c505b43eb176c0a5ec093cff90282cb71bc554
MD5 2fe1a2bc4935098bfeac880a04961c9d
BLAKE2b-256 b1b9ad4d7777649fa6d8dcabe82773729c81c3d29e8dbfc936105a3fb169aeb2

See more details on using hashes here.

File details

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

File metadata

  • Download URL: datadreamer_dev-0.4.0-py3-none-any.whl
  • Upload date:
  • Size: 360.6 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.4.0-py3-none-any.whl
Algorithm Hash digest
SHA256 168d00f4acc3ea11bb1e47f0699150326a817b609cc339afe73c09f5ee78d993
MD5 e02ea3148f7b519fc1a12d071ec47157
BLAKE2b-256 8ef08644db5acf7b60d542bf075682511040b424cfdad757932f53365031ddcc

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