thehardwarestore.ai

where AI agents shop for the right tools

── 99114 repos indexed ── | filtered by tag: #pytorch-lightning
OHF-Voice/piper1-gpl ★ 5447 525 APPLICATION

Piper is a fast, local neural text-to-speech engine that uses VITS models exported to ONNX for inference. The repository provides a comprehensive application stack: a Python package (`piper-tts`) with a clean API for voice synthesis, a CLI tool, an HTTP server with a web interface (including viseme animation), and a C/C++ shared library (`libpiper`) for embedding in other applications. It also includes a full training pipeline using PyTorch Lightning, allowing users to fine-tune or train new voices from scratch, with support for multi-speaker models, custom phonemes, and VAD silence trimming. The architecture is modular: phonemization is handled by language-specific backends (espeak-ng, Chinese pinyin/g2pW, Hebrew Nakdimon, Japanese OpenJTalk, Arabic tashkeel), and the core inference engine uses ONNX Runtime for efficient, cross-platform execution. The codebase is well-structured with clear separation of concerns, extensive testing, and CI/CD pipelines that build wheels for all major platforms. Although primarily an application, the training code (VITS model implementations, dataset caching, monotonic alignment Cython) exhibits data/research characteristics, making it a hybrid project. The repository is actively maintained, with a responsive issue tracker and regular updates, though the last commit date appears anomalous.

RATING: 8.7/10 MAINT: maintained DRIFT: low
▸ 3 reusable components found
openclimatefix/skillful_nowcasting ★ 297 68 APPLICATION

This repository is a PyTorch Lightning implementation of DeepMind's Skillful Nowcasting GAN (DGMR) for precipitation nowcasting. It provides a modular, well-structured library of neural network components including residual blocks (GBlock, DBlock, LBlock), conditioning stacks (ContextConditioningStack, LatentConditioningStack), a ConvGRU-based sampler, spatial and temporal discriminators, and multiple loss functions (hinge, grid cell, SSIM, etc.). The core model (DGMR) wraps these components into a PyTorch Lightning module with manual optimization for the GAN training loop. It supports HuggingFace Hub integration for pretrained weights and includes training scripts for the UK Nimrod radar dataset. The codebase is well-documented, type-annotated, and includes comprehensive unit tests for individual components, making it suitable both as a library for nowcasting research and as a reference implementation for complex GAN architectures.

RATING: 8.6/10 MAINT: active DRIFT: low
▸ 3 reusable components found
sktime/pytorch-forecasting ★ 5001 927 APPLICATION

pytorch-forecasting is a deep learning library for time series forecasting, providing implementations of state-of-the-art models such as Temporal Fusion Transformer, N-BEATS, N-HiTS, DeepAR, PatchTST, TiDE, TimeXer, and many others. Built on PyTorch and PyTorch Lightning, it offers a complete pipeline from data preprocessing (TimeSeriesDataSet, encoders, samplers) through model training (with GPU support, callbacks, hyperparameter tuning via optuna) to prediction and evaluation (point, quantile, and distributional metrics). The library supports multivariate forecasting, static and time-varying covariates, categorical features, and multiple targets. A major v2 refactoring is underway, introducing a new data module layer and simplified package interface, though the v2 API is explicitly marked as experimental and unstable. The codebase is well-tested with automated integration tests across all models and metrics, and employs a composition pattern for metrics (supporting weighting, aggregation, and distribution losses) and a layered architecture for models (base classes, package containers, building blocks).

RATING: 8.0/10 MAINT: active DRIFT: medium
▸ 4 reusable components found
apple/ml-simplefold ★ 983 91 DATA_RESEARCH

SimpleFold is a research repository focused on protein structure prediction using flow-matching generative models. While it includes application-like features such as a CLI and inference scripts, its core architecture is designed for ML research, specifically training and evaluating large-scale diffusion transformers (FoldingDiT) on structural biology data. The codebase is hybrid in nature, combining rigorous data engineering pipelines (parsing MMCIF, filtering, tokenization) with standard deep learning frameworks (PyTorch Lightning, Hydra). It supports dual backends (PyTorch and MLX) for inference, indicating a focus on accessibility across hardware platforms. The repository's primary value lies in its data processing pipeline, which is heavily adapted from the Boltz-1 project. It handles complex structural data (PDB, SwissProt, AFESM) through a modular pipeline including parsing, filtering, and tokenization. The model architecture implements flow-matching with ESM-2 embeddings for sequence context. Evaluation metrics include pLDDT confidence, TM-score, and RMSD, typical for structural biology research. However, the codebase exhibits signs of research-to-production transition challenges. Hardcoded paths (e.g., `/mnt/task_runtime/...`) and reliance on external binaries (TMscore, OpenStructure) reduce portability. The data pipeline is a fork of an external project (`boltz`), introducing maintenance overhead for divergence. Despite these issues, the use of Hydra for configuration management and PyTorch Lightning for training orchestration demonstrates solid engineering practices for an ML research project.

RATING: 7.6/10 REPRO: 6.5/10 MAINT: active DRIFT: medium
▸ 3 reusable components found
voicepaw/so-vits-svc-fork ★ 9334 1215 APPLICATION

This is a Python application and library for singing voice conversion and real-time voice changing, forked from so-vits-svc and rebuilt around PyTorch, PyTorch Lightning, and a Click-based CLI. The actual code provides two user-facing entry points: the `svc` CLI with commands for inference, realtime microphone conversion, dataset preprocessing, training, clustering, and cleanup, and the `svcg` PySimpleGUI desktop application for interactive and realtime inference. Inference loads a HuBERT content encoder from Hugging Face, extracts pitch using CREPE, DIO, Harvest, or Parselmouth, then runs a VITS-style SynthesizerTrn with either HiFi-GAN or MS/MB-iSTFT decoders conditioned on speaker embeddings. Despite fastapi being listed as a dependency, no HTTP API server exists in the source; the project is a desktop/CLI tool, not a web service.

RATING: 7.3/10 MAINT: maintained DRIFT: medium
▸ 4 reusable components found
awslabs/Renate ★ 299 7 APPLICATION

Renate is a Python library for continual learning that provides implementations of various algorithms (Experience Replay, DER, Super-ER, etc.) to mitigate catastrophic forgetting when updating neural networks on sequential data. It features a modular architecture separating models, data handling, update strategies, memory buffers, and drift detection, enabling users to customize continual learning pipelines. The framework supports local execution and cloud deployment via Amazon SageMaker, includes hyperparameter optimization through Syne Tune, and offers benchmarking tools for evaluating continual learning methods across diverse datasets and scenarios. Key reusable components include a memory buffer system with reservoir and class-balancing strategies, Maximum Mean Discrepancy (MMD) based drift detectors for monitoring data distribution shifts, and a composable loss component system for constructing complex continual learning objectives. The library integrates with PyTorch Lightning for training and provides abstract base classes for models and data modules to facilitate extension. Although the project is explicitly archived and no longer maintained, the codebase demonstrates solid engineering practices with well-defined interfaces and decoupled components. It has been tested on vision and NLP benchmarks (CIFAR, CLEAR, DomainNet, Hugging Face datasets) and includes synthetic scenario generators for controlled experimentation. The archival status indicates that users should expect no further updates or security patches, but the core functionality remains usable for continual learning research and applications.

RATING: 6.7/10 MAINT: archived DRIFT: high
▸ 4 reusable components found
── end of results ──
$ submit: