Repository Index

A standard API for single-agent reinforcement learning environments, with popular reference environments and related utilities (formerly Gym)

• api
• gym
• reinforcement-learning

Transformer related optimization, including BERT, GPT

• pytorch
• transformer
• gpt
• bert

docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning. Ongoing development and maintenance by t2k.

• ocr
• deep-learning
• document-recognition
• text-detection

A standard API for multi-agent reinforcement learning environments, with popular reference environments and related utilities

• api
• gymnasium
• multi-agent-reinforcement-learning
• reinforcement-learning

Pretrained language model and its related optimization techniques developed by Huawei Noah's Ark Lab.

• knowledge-distillation
• model-compression
• quantization
• pretrained-models

We unified the interfaces of instruction-tuning data (e.g., CoT data), multiple LLMs and parameter-efficient methods (e.g., lora, p-tuning) together for easy use. We welcome open-source enthusiasts to initiate any meaningful PR on this repo and integrate as many LLM related technologies as possible. 我们打造了方便研究人员上手和使用大模型等微调平台，我们欢迎开源爱好者发起任何有意义的pr！

• chatglm
• llama
• llm
• lora

Paper and implementation of UNet-related model.

• segmentation
• unet
• pytorch

This package contains deep learning models and related scripts for RoseTTAFold

Nodes related to video workflows

A project demonstrating Lidar related AI solutions, including three GPU accelerated Lidar/camera DL networks (PointPillars, CenterPoint, BEVFusion) and the related libs (cuPCL, 3D SparseConvolution, YUV2RGB, cuOSD,).

A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities

• datasets
• gymnasium
• offline-rl
• reinforcement-learning

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

• eeg
• deep-learning
• convolutional-neural-networks
• keras

A PyTorch library entirely dedicated to neural differential equations, implicit models and related numerical methods

• deep-learning
• neural-network
• neural-differential-equations
• pytorch

Support vector machines (SVMs) and related kernel-based learning algorithms are a well-known class of machine learning algorithms, for non-parametric classification and regression. liquidSVM is an implementation of SVMs whose key features are: fully integrated hyper-parameter selection, extreme speed on both small and large data sets, full flexibility for experts, and inclusion of a variety of different learning scenarios: multi-class classification, ROC, and Neyman-Pearson learning, and least-squares, quantile, and expectile regression.

• svm
• regression
• classification
• machine-learning