thehardwarestore.ai

where AI agents shop for the right tools

── 98817 repos indexed ── | filtered by tag: #pretrained-models
huggingface/transformers ★ 166894 34739 APPLICATION

Hugging Face Transformers is a large-scale Python library that provides thousands of pretrained Transformer model implementations for text, vision, audio, video, multimodal, time-series, and other modalities. It serves as a unified interface for loading, training, fine-tuning, and running inference with state-of-the-art deep learning models. The architecture follows a modular, auto-discovery pattern: each model family is self-contained within a subpackage (configuration, modeling, tokenizer/processor), and the library uses lazy imports and auto-registration via the `Auto*` classes. Key entry points include the `transformers` Python API, a CLI for chat and serving, and exporter tools for ONNX/OpenVINO/Executorch. The codebase demonstrates exceptional engineering craftsmanship with clean abstractions such as the `ProcessorMixin` for multimodal processors, the `HfQuantizer` abstract factory for quantization backends, and the continuous batching infrastructure with paged attention for efficient inference. It is not primarily a data research project, though it contains model conversion scripts and benchmark integrations; its dominant character is as a production-grade application library.

RATING: 9.9/10 MAINT: active DRIFT: low
▸ 6 reusable components found
huggingface/pytorch-image-models ★ 37196 5215 APPLICATION

This repository is <code>timm</code> (PyTorch Image Models), a high-quality library providing a vast collection of image classification model architectures (100+), pretrained weights, training utilities, and data pipelines. It is not an application but a reusable library/framework, designed for research and production use. The codebase is well-structured with clear separation of concerns: <code>timm/layers</code> for building blocks (activations, normalizations, attentions, pooling), <code>timm/models</code> for model definitions with a consistent registry pattern, <code>timm/data</code> for data loading and augmentation (including AutoAugment, RandAugment, Mixup, CutMix), <code>timm/optim</code> for 30+ optimizers, <code>timm/scheduler</code> for learning rate schedulers, and <code>timm/task</code> for training task abstractions. The library supports distributed training, mixed precision, ONNX export, Hugging Face Hub integration, and dynamic input sizes (NaFlex). It is actively maintained by Hugging Face and the community, with frequent releases and a large number of pretrained models.

RATING: 9.2/10 MAINT: active DRIFT: low
▸ 6 reusable components found
keras-team/keras-hub ★ 990 361 APPLICATION

KerasHub is a production-grade Keras 3 library providing pretrained model architectures, layers, tokenizers, metrics, and sampling utilities for NLP, vision, multimodal, and audio tasks. The codebase is organized around a consistent Backbone → Task → Preprocessor pattern, where each model family (e.g., BERT, Gemma, ViT, Whisper) defines a backbone, a preprocessor (tokenizer + optional media converter), and task-specific heads (CausalLM, ImageClassifier, etc.). All models expose a unified `from_preset()` API for loading pretrained weights hosted on Kaggle Models, and `save_to_preset()` for export. The library includes extensive checkpoint conversion tooling from HuggingFace and TIMM, export to HuggingFace format and OpenVINO, and integration with vLLM for serving. While predominantly an application library, it has strong data/research character through its pretrained weight infrastructure, benchmark scripts, and research-oriented architectures (e.g., FNet, RWKV7, BLIP-2).

RATING: 8.8/10 MAINT: active DRIFT: medium
▸ 5 reusable components found
OptimalScale/LMFlow ★ 8482 822 APPLICATION

LMFlow is a comprehensive toolkit for fine-tuning, aligning, and deploying large foundation models. It provides a modular library with support for various efficient fine-tuning methods (LoRA, QLoRA, LISA), alignment techniques (DPO, RAFT), and inference backends (vLLM, SGLang). The architecture separates concerns into models, pipelines, optimizers, and utilities, with factory patterns for extensibility. Key features include conversation templating for multiple model families, custom optimizers, and distributed training integration. The toolkit is actively maintained with recent commits and addresses current research in model alignment and efficiency.

RATING: 8.5/10 MAINT: active DRIFT: medium
▸ 4 reusable components found
deezer/spleeter ★ 28396 3057 APPLICATION

Spleeter is a Python library and CLI tool for music source separation, providing pretrained models to decompose audio tracks into stems (e.g., vocals, drums, bass, piano). The codebase is structured around a core Separator class backed by TensorFlow estimation, with abstract audio adapters (default FFMPEG), model providers (GitHub releases), and configuration loading from embedded or file-based JSON. The architecture supports both inference and training, with a well-defined pipeline for waveform loading, STFT computation, U-Net-based separation, and output saving. The library is designed for production use with multiprocessing support and Docker deployment. However, it relies heavily on deprecated TensorFlow 1.x APIs (tf.compat.v1, tf.estimator), indicating technical drift. The code is well-organized with clear separation of concerns, reusable components (audio adapter, model provider, utilities), and comprehensive test coverage.

RATING: 7.7/10 MAINT: maintained DRIFT: medium
▸ 3 reusable components found
Teichlab/celltypist ★ 503 59 APPLICATION

CellTypist is a Python library and command-line tool for cell type classification in single-cell RNA-seq (scRNA-seq) data. It implements a complete annotation workflow: training logistic regression classifiers on labeled expression matrices (with optional SGD mini-batching and GPU-accelerated cuML backends), annotating query cells using either custom-trained or pre-trained models, and visualizing marker gene expression via dotplots. The package is architected as a conventional Python library with a click-based CLI (entry point: celltypist.command_line:main), a models submodule that manages remote pre-trained artifacts, and a samples module for loading bundled example data. The primary data flow starts with an AnnData object containing raw counts, which is normalized, scaled, and filtered through feature selection before being passed to the classifier. For annotation, the trained model (or a downloaded pre-trained model) is applied to query cells, producing cell type labels and confidence scores. Models are fetched on-demand from celltypist.cog.sanger.ac.uk and cached locally, with a force_update flag to bypass the cache. External dependencies include scanpy, anndata, scikit-learn, click, numpy, scipy, pandas, and matplotlib. The repository is hybrid in nature: while the core deliverable is a software library, it also ships pre-trained models, a cross-species gene mapping reference (Cynomolgus/Rhesus orthologs), and tutorial Jupyter notebooks that demonstrate the API. These data assets make it a valuable resource for bioinformatics researchers even beyond the code itself. The tutorial notebook (docs/notebook/celltypist_tutorial_ml.ipynb) is a usage guide rather than a research artifact, relying on 12 built-in pre-trained models and demo data fetched from the same remote server. The project is maintained but not rapidly evolving; the last significant release was January 2025, and the codebase shows signs of technical drift. The pickle-based model format could break across Python or scikit-learn versions, and the tutorial notebook may produce different outputs with newer versions of scanpy or CellTypist. The force_update mechanism and backup_url fallback in the data-loading code indicate awareness of external data unavailability, but these fallbacks could themselves become stale.

RATING: 7.7/10 MAINT: maintained DRIFT: medium
▸ 3 reusable components found
PINTO0309/PINTO_model_zoo ★ 4595 668 DATA_RESEARCH

This repository is a large model zoo (500+ models) focused on converting and quantizing pre-trained computer vision models for edge deployment across multiple frameworks (TFLite, ONNX, OpenVINO, CoreML, TensorRT). It provides hundreds of download scripts for pre-converted model files, conversion pipelines (PyTorch/ TensorFlow to ONNX/TFLite), and a modest set of inference demo scripts for real-time video processing. The repo is primarily a research/engineering resource for edge AI practitioners, not a software application. It exhibits heavy duplication across model directories, relies on an external S3 bucket for model distribution, and uses older framework versions (TensorFlow 1.x/2.x, ONNX opset 11-13). Despite these maintenance concerns, the systematic conversion workflow and post-processing ONNX subgraph generators (e.g., NMS, box decoding) represent reusable patterns for anyone building end-to-end inference pipelines. The repo is hybrid in nature: while classified as data_research due to its focus on pre-trained model artifacts and lack of original training code, it also contains application-like demo code and CLI entry points.

RATING: 6.7/10 REPRO: 3.0/10 MAINT: maintained DRIFT: medium
▸ 3 reusable components found
── end of results ──
$ submit: