Skip to content
项目分类

机器学习 · GitHub

发现模型训练、评估与推理框架,比较深度学习和机器学习项目的社区关注度。

实用 GitHub 项目

4,743 个项目
1,591星标+324小时

This is the Army Research Laboratory (ARL) EEGModels Project: A Collection of Convolutional Neural Network (CNN) models for EEG signal classification, using Keras and Tensorflow

wespeaker

wenet-e2e

1,467星标+324小时

Research and Production Oriented Speaker Verification, Recognition and Diarization Toolkit

机器学习PythonApache-2.0GitHub

NamiGPT

aa1143

218星标+224小时
NamiGPT preview

零基础入门大模型(LLM)的基础教程:从零手写 GPT-2,一行行看懂大模型是怎么工作的 | Build GPT-2 from scratch in PyTorch, step by step

RAVE

acids-ircam

1,831星标+224小时
RAVE preview

Official implementation of the RAVE model: a Realtime Audio Variational autoEncoder

lama

advimman

10,381星标+224小时
lama preview

🦙 LaMa 图像修复:基于傅里叶卷积的分辨率鲁棒大掩码图像修复,WACV 2022

机器学习Jupyter NotebookApache-2.0GitHub
1,763星标+224小时
ComfyUI_VNCCS preview

Visual Novel Character Creation Suite is a comprehensive tool for creating character sprites for visual novels. It allows you to create unique characters with a consistent appearance across all images, which was previously a challenging task when using neural networks.

机器学习JavaScriptMITGitHub

NN-SVG

alexlenail

5,701星标+224小时
NN-SVG preview

可直接用于发表的神经网络架构示意图。

机器学习JavaScriptMITGitHub

torchcurves

alexshtf

72星标+224小时
torchcurves preview

Parametric differentiable curves with PyTorch for continuous embeddings, shape-restricted models, or KANs

  • Differentiable: Custom autograd function ensures gradients flow properly through the curve evaluation.
  • Vectorized: Vectorized operations for efficient batch and multi-curve evaluation.
机器学习PythonApache-2.0GitHub

MLQuestions

andrewekhalel

4,965星标+224小时

Machine Learning and Computer Vision Engineer - Technical Interview Questions

  • Ridge (L2 norm)
  • Oversampling or undersampling. Instead of sampling with a uniform distribution from the training dataset, we can use other distributions so the model sees a more balanced dataset.
812星标+224小时
fast-plate-ocr preview

Lightweight & fast OCR models for license plate text recognition.

  • Keras 3 Backend Support: Train seamlessly using TensorFlow, JAX, or PyTorch backends 🧠
  • Efficient Execution: Lightweight models that are cheap to run 💰

voxlock

arbazsiddiqui

110星标+224小时
voxlock preview

The best performing voice isolator there is. Keeps the caller's voice and removes the other voices in the room before speech-to-text hears them. Open source, streaming, runs on CPU and in the browser. pip install voxlock

机器学习PythonApache-2.0GitHub

rosnav-rl

Arena-Rosnav

106星标+224小时

Modular DRL framework for autonomous robot navigation in ROS2. Plug-and-play RL backends (Stable-Baselines3, DreamerV3), composable reward functions, observation spaces & neural architectures - built for research and deployment.

  • Wrap any RL backend - the common RL_Model interface means you swap SB3 ↔ DreamerV3 ↔ your own implementation without touching training code. Same observations, same reward, same config.
  • Declarative data pipeline - define collectors by ROS message type (sensor_msgs/LaserScan); preprocessing and topic wiring happen automatically.

pytorch

Ascend

607星标+224小时
pytorch preview

As the core component of the Ascend for PyTorch community, TorchNPU is a deep learning adapter plugin built by Ascend for PyTorch, enabling the PyTorch framework to directly invoke Ascend NPU and providing developers with the powerful compute capabilities of Ascend AI processors. (About Ascend NPU: Ascend NPU (Neural Processing Unit) is a series of

  • Basic Compute: Extensive support for PyTorch native APIs and custom APIs, covering mainstream AI scenarios with a consistent experience, enabling rapid model and algorithm implementation.
  • Distributed: Supports accelerating distributed training via FSDP2. Core compute APIs support DTensor, along with collective communication primitives such as AllGather, AllReduce, and AllToAll, as well as point-to-point primitives like Send and Recv.