pgml

pgml (PostgresML): The open-source PostgreSQL extension that transforms your database into a complete, in-database AI/ML platform, executing training and inference directly via SQL.

pgml is the PostgresML extension, a powerful tool for building AI applications directly within PostgreSQL. The core principle: Move models to the data, not data to the models. This eliminates costly data transfers, drastically improving latency and efficiency for ML workflows. Users leverage standard SQL to train, deploy, and run models (Scikit-learn, XGBoost, LLMs) on their existing data. It supports GPU-powered inference, integrates with Hugging Face for state-of-the-art NLP tasks, and transforms Postgres into a vector database for Retrieval-Augmented Generation (RAG) applications. This platform simplifies the entire MLOps pipeline, making the database the central hub for all data and machine learning operations.

https://postgresml.org