Production-ready
data processing made
easy
and
shareable
Explore the docs »
🚀 Production-ready | 👶 Easy | 👫 Shareable |
---|---|---|
Benefit from built-in features such as autoscaling, data lineage, and pipeline caching, and deploy to (managed) platforms such as Vertex AI, Sagemaker, and Kubeflow Pipelines. | Implement your custom data processing code using datastructures you know such as Pandas dataframes. Move from local development to remote deployment without any code changes. |
Fondant components are defined by a clear interface, which makes them reusable and shareable. Compose your own pipeline using components available on our hub. |
With the advent of transfer learning and now foundation models, everyone has started sharing and reusing machine learning models. Most of the work now goes into building data processing pipelines, which everyone still does from scratch. This doesn't need to be the case, though, if processing components would be shareable and pipelines composable. Realizing this is the main vision behind Fondant.
Towards that end, Fondant offers:
- 🔧 Plug ‘n’ play composable data processing pipelines
- 🧩 Library containing off-the-shelf reusable components
- 🐼 A simple Pandas based interface for creating custom components
- 📊 Built-in lineage, caching, and data explorer
- 🚀 Production-ready, scalable deployment
- ☁️ Integration with runners across different clouds (Vertex, Sagemaker, Kubeflow)
Eager to get started? Follow our step by step guide to get your first pipeline up and running.
Fondant comes with a library of reusable components that you can leverage to compose your own pipeline:
- Data ingestion: S3, GCS, ABS, Hugging Face, local file system, ...
- Data Filtering: Duplicates, language, visual style, topic, format, aesthetics, NSFW, license, ...
- Data Enrichment: Captions, segmentations, embeddings, ...
- Data Transformation: Image cropping, image resizing, text chunking, ....
- Data retrieval: Common Crawl, LAION, ...
👉 Check our Component Hub for an overview of all available components
We have created several ready-made example pipelines for you to use as a starting point for exploring Fondant. They are hosted as separate repositories containing a notebook tutorial so you can easily clone them and get started:
📖 RAG tuning pipeline
End-to-end Fondant pipelines to index and evaluate RAG (Retrieval-Augmented Generation) systems.
🛋️ ControlNet Interior Design Pipeline
An end-to-end Fondant pipeline to collect and process data for the fine-tuning of a ControlNet model, focusing on images related to interior design.
🖼️ Filter creative common license images
An end-to-end Fondant pipeline that starts from our Fondant-CC-25M creative commons image dataset and filters and downloads the desired images.
🔢 Datacomp pipeline
An end-to-end Fondant pipeline filtering image-text data to train a CLIP model for the DataComp competition.
Fondant can be installed using pip:
pip install fondant
For more detailed installation options, check the installation page on our documentation.
Fondant allows you to easily define data pipelines comprised of both reusable and custom
components. The following pipeline for instance uses the reusable load_from_hf_hub
component
to load a dataset from the Hugging Face Hub and process it using a custom component:
pipeline.py
from fondant.pipeline import Pipeline
pipeline = Pipeline(pipeline_name="example pipeline", base_path="fs://bucket")
dataset = pipeline.read(
"load_from_hf_hub",
arguments={
"dataset_name": "lambdalabs/pokemon-blip-captions"
},
)
dataset = dataset.apply(
"components/custom_component",
arguments={
"min_width": 600,
"min_height": 600,
},
)
To create a custom component, you first need to describe its contract as a yaml specification. It defines the data consumed and produced by the component and any arguments it takes.
fondant_component.yaml
name: Custom component
description: This is a custom component
image: custom_component:latest
consumes:
image:
type: binary
produces:
caption:
type: utf8
args:
argument1:
description: An argument passed to the component at runtime
type: str
argument2:
description: Another argument passed to the component at runtime
type: str
Once you have your component specification, all you need to do is implement a constructor
and a single .transform
method and Fondant will do the rest. You will get the data defined in
your specification partition by partition as a Pandas dataframe.
component/src/main.py
import pandas as pd
from fondant.component import PandasTransformComponent
class ExampleComponent(PandasTransformComponent):
def __init__(self, *, argument1, argument2, **kwargs) -> None:
"""
Args:
argumentX: An argument passed to the component
"""
# Initialize your component here based on the arguments
def transform(self, dataframe: pd.DataFrame) -> pd.DataFrame:
"""Implement your custom logic in this single method
Args:
dataframe: A Pandas dataframe containing the data
Returns:
A pandas dataframe containing the transformed data
"""
For more advanced use cases, you can use the DaskTransformComponent
instead.
Once you have a pipeline you can easily run (and compile) it by using the built-in CLI:
fondant run local pipeline.py
To see all available runner and arguments you can check the fondant CLI help pages
fondant --help
Or for a subcommand:
fondant <subcommand> --help
We welcome contributions of different kinds:
Issues | If you encounter any issue or bug, please submit them as a Github issue. You can also submit a pull request directly to fix any clear bugs. |
Suggestions and feedback | Our roadmap and priorities are defined based on community feedback. To provide input, you can join our discord or submit an idea in our Github Discussions. |
Framework code contributions | If you want to help with the development of the Fondant framework, have a look at the issues marked with the good first issue label. If you want to add additional functionality, please submit an issue for it first. |
Reusable components | Extending our library of reusable components is a great way to contribute. If you built a component which would be useful for other users, please submit a PR adding them to the components/ directory. You can find a list of possible contributable components here or your own ideas are also welcome! |
For a detailed view on the roadmap and day to day development, you can check our github project board.
You can also check out our architecture page to familiarize yourself with the Fondant architecture and repository structure.
We use poetry and pre-commit to enable a smooth developer flow. Run the following commands to set up your development environment:
pip install poetry
poetry install --all-extras
pre-commit install