superduper-io / superduper

Superduper: Integrate AI models and machine learning workflows with your database to implement custom AI applications, without moving your data. Including streaming inference, scalable model hosting, training and vector search.
https://superduper.io
Apache License 2.0
4.65k stars 448 forks source link
ai chatbot data database distributed-ml inference llm-inference llm-serving llmops ml mlops mongodb pretrained-models python pytorch rag semantic-search torch transformers vector-search
SuperDuper logo

Bring AI to your favourite database

Docs | Blog | Website | Use-Cases | Slack | Youtube | LinkedIn

Package version Supported Python versions License - Apache 2.0

What is Superduper?

Superduper (formerly SuperDuperDB) is a Python framework for integrating AI models and workflows with major databases. Implement custom AI solutions without moving your data through complex pipelines and specialized vector databases, including hosting of your own models, streaming inference and scalable model training/fine-tuning.

Transform your existing database into an AI development and deployment stack with one command, streamlining your AI workflows in one environment instead of being spread across systems and environments:

db = superduper('mongodb|postgres|mysql|sqlite|duckdb|snowflake://<your-db-uri>')

Run Superduper anywhere, or contact us to learn more about the enterprise platform for bringing your apps to production at scale.

Key features

Preview

Browse the re-usable snippets to understand how to accomplish difficult AI end-functionality with few lines of code using Superduper.

Example use-cases and apps (notebooks)

The notebooks below are examples how to make use of different frameworks, model providers, databases, retrieval techniques and more. To learn more about how to use Superduper with your database, please check our Docs.

| Name | Link | |-------------------------------------------------|---------------------------------------------------------------------------------------------------------------------------------------------------| | Multimodal vector-search with a range of models and datatypes | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/superduper-io/superduper/blob/main/docs/content/use_cases/multimodal_vector_search_image.ipynb) | | RAG with self-hosted LLM | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://github.com/superduper-io/superduper/blob/main/docs/content/use_cases/retrieval_augmented_generation.ipynb) | | Fine-tune an LLM on your database | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://github.com/superduper-io/superduper/blob/main/docs/content/use_cases/fine_tune_llm_on_database.ipynb) | | Featurization and transfer learning | [![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://github.com/superduper-io/superduper/blob/main/docs/content/use_cases/transfer_learning.ipynb) |
## Currently supported datastores: Superduper your: [MongoDB](https://www.mongodb.com), [MongoDB Atlas](https://www.mongodb.com/cloud/atlas), [Snowflake](https://www.snowflake.com), [PostgreSQL](https://www.postgresql.org), [MySQL](https://www.mysql.com), [SQLite](https://www.sqlite.org), [DuckDB](https://duckdb.org), [Google BigQuery](https://cloud.google.com/bigquery), [Amazon S3](https://aws.amazon.com/s3/), [Microsoft SQL Server (MSSQL)](https://www.microsoft.com/en-us/sql-server), [ClickHouse](https://clickhouse.com), [Oracle](https://www.oracle.com/database/), [Trino](https://trino.io), [PySpark](https://spark.apache.org/docs/latest/api/python/), [Pandas](https://pandas.pydata.org), [Apache Druid](https://druid.apache.org), [Apache Impala](https://impala.apache.org), [Polars](https://www.pola.rs), [Apache Arrow DataFusion](https://arrow.apache.org/datafusion/), ## Supported AI frameworks, models and APIs (*more coming soon*): Integrate and self-hosted your own models (whether from open-source, commercial or self-developed) with a simple Python command from: [PyTorch](https://pytorch.org), [Scikit-learn](https://scikit-learn.org), [HuggingFace](https://huggingface.co) ## Preconfigured API integrations (*more coming soon*): Integrate externally hosted models accessible via API to work side-by-side or together with your other models a simple Python command: [OpenAI](https://www.openai.com), [Cohere](https://cohere.ai), [Anthropic](https://www.anthropic.com), [Jina AI](https://jina.ai) ## Installation #### # Option 1. Superduper Library Ideal for building new AI applications. ```shell pip install superduper-framework ``` #### # Option 2. Superduper Container Ideal for learning basic Superduper functionalities and testing notebooks. ```shell docker pull superduperio/superduper docker run -p 8888:8888 superduperio/superduper ``` #### # Option 3. Superduper Testenv Ideal for learning advanced Superduper functionalities and testing whole AI stacks. ```shell make build_sandbox make testenv_init ``` ## Community & getting help If you have any problems, questions, comments, or ideas: - Join our Slack (we look forward to seeing you there). - Search through our GitHub Discussions, or add a new question. - Comment an existing issue or create a new one. - Help us to improve Superduper by providing your valuable feedback here! - Email us at `gethelp@superduper.io`. - Visit our [YouTube channel](https://www.youtube.com/@superduper-io). - Follow us on [Twitter (now X)](https://twitter.com/superduperdb). - Connect with us on [LinkedIn](https://www.linkedin.com/company/superduper-io). - Feel free to contact a maintainer or community volunteer directly! ## Contributing There are many ways to contribute, and they are not limited to writing code. We welcome all contributions such as: - Bug reports - Documentation improvements - Enhancement suggestions - Feature requests - Expanding the tutorials and use case examples Please see our [Contributing Guide](CONTRIBUTING.md) for details. ## Contributors Thanks goes to these wonderful people: ## License Superduper is open-source and intended to be a community effort, and it wouldn't be possible without your support and enthusiasm. It is distributed under the terms of the Apache 2.0 license. Any contribution made to this project will be subject to the same provisions. ## Join Us We are looking for nice people who are invested in the problem we are trying to solve to join us full-time. Find roles that we are trying to fill here!