Pretrain, finetune ANY AI model of ANY size on 1 or 10,000+ GPUs with zero code changes.

The world's simplest facial recognition api for Python and the command line

Datasets, Transforms and Models specific to Computer Vision

NLTK Source

A hyperparameter optimization framework

🐍 Geometric Computer Vision Library for Spatial AI

A python library for user-friendly forecasting and anomaly detection on time series.

Accessible large language models via k-bit quantization for PyTorch.

Uniform Manifold Approximation and Projection

Improve your resumes with Resume Matcher. Get insights, keyword suggestions and tune your resumes to job descriptions.

🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Deep Learning for humans

We write your reusable computer vision tools. 💜

Tensors and Dynamic neural networks in Python with strong GPU acceleration

PyTorch version of Stable Baselines, reliable implementations of reinforcement learning algorithms.

Kedro is a toolbox for production-ready data science. It uses software engineering best practices to help you create data engineering and data science pipelines that are reproducible, maintainable, and modular.