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.6.0.tar.gz (272.8 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.6.0-py3-none-any.whl (360.8 kB view details)

Uploaded Python 3

File details

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

File metadata

  • Download URL: datadreamer_dev-0.6.0.tar.gz
  • Upload date:
  • Size: 272.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: poetry/1.7.1 CPython/3.11.7 Darwin/23.1.0

File hashes

Hashes for datadreamer_dev-0.6.0.tar.gz
Algorithm Hash digest
SHA256 43acfd6fa1d2164a8f75a15cf026daf375ed3e74c1bd8740ff2877069986303b
MD5 a9902c490ebb3ae72808a1b6b39f0917
BLAKE2b-256 01e36173cf025e1cbc51f81c00399f88fc85dc4c4eacdb3bd17be21584add646

See more details on using hashes here.

File details

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

File metadata

  • Download URL: datadreamer_dev-0.6.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.11.7 Darwin/23.1.0

File hashes

Hashes for datadreamer_dev-0.6.0-py3-none-any.whl
Algorithm Hash digest
SHA256 38934aeee8e965ad8e5147de4902cf43aca15417d52a30ec40f02df28c4a9d18
MD5 8a9932d5f17a10ce551052bc2ff2e794
BLAKE2b-256 aa039338b84c72b216f5ab9ffbc80938561f4d6187f16691de1d75ddc31df9c2

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