Repository Index

The most powerful and modular diffusion model GUI, api and backend with a graph/nodes interface. The fastest local inference engine in the world.

• stable-diffusion
• pytorch
• ai
• python

Port of OpenAI's Whisper model in C/C++

• openai
• speech-to-text
• transformer
• whisper

Algorithm powering the For You feed on X

JARVIS, a system to connect LLMs with ML community. Paper: https://arxiv.org/pdf/2303.17580.pdf

• deep-learning
• platform
• pytorch

A fast, distributed, high performance gradient boosting (GBT, GBDT, GBRT, GBM or MART) framework based on decision tree algorithms, used for ranking, classification and many other machine learning tasks.

• gbdt
• gbm
• machine-learning
• data-mining

ComfyUI-Manager is an extension designed to enhance the usability of ComfyUI. It offers management functions to install, remove, disable, and enable various custom nodes of ComfyUI. Furthermore, this extension provides a hub feature and convenience functions to access a wide range of information within ComfyUI.

Tensor library for machine learning

• machine-learning

Style and Grammar Checker for 25+ Languages

• grammar
• natural-language
• natural-language-processing
• style-checker

text and image to video generation: CogVideoX (2024) and CogVideo (ICLR 2023)

• llm
• sora
• video-generation
• cogvideox

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

• segmentation
• image-processing
• pspnet
• unet

:robot: :speech_balloon: Deep learning for Text to Speech (Discussion forum: https://discourse.mozilla.org/c/tts)

• deep-learning
• text-to-speech
• python
• pytorch

A logical, reasonably standardized, but flexible project structure for doing and sharing data science work.

• cookiecutter-data-science
• cookiecutter
• cookiecutter-template
• data-science

An open source reinforcement learning framework for training, evaluating, and deploying robust trading agents.

Gorgonia is a library that helps facilitate machine learning in Go.

• machine-learning
• artificial-intelligence
• neural-network
• computation-graph

Sacred is a tool to help you configure, organize, log and reproduce experiments developed at IDSIA.

• python
• machine-learning
• infrastructure
• reproducible-research

Make videos on the computer you already own. FreeVideo runs MiniMax H3 in as little as 8 GB of VRAM and 16 GB of RAM, and adapts its acceleration path to your hardware.

• diffusion-models
• inference
• minimax-h3
• video-generation

Assist in organizing your piles of documents, resulting from scanners, e-mails and other sources with miminal effort.

• ocr
• stanford-corenlp
• personal-document-system
• scala

PGPortfolio: Policy Gradient Portfolio, the source code of "A Deep Reinforcement Learning Framework for the Financial Portfolio Management Problem"(https://arxiv.org/pdf/1706.10059.pdf).

Machine Learning in R

• machine-learning
• data-science
• tuning
• cran

Official Project Page for Recurrent Looped Transformer (RLT) (https://arxiv.org/abs/2610.07591)

mlr3: Machine Learning in R - next generation

• machine-learning
• data-science
• classification
• regression

onnx-go gives the ability to import a pre-trained neural network within Go without being linked to a framework or library.

• go
• onnx
• neural-network
• open-source

RNNLIB is a recurrent neural network library for sequence learning problems. Forked from Alex Graves work http://sourceforge.net/projects/rnnl/

Official implementation of “UniForge3D: Joint 3D Reconstruction and Generation in a Shared Anchor-Guided Space.”