ZhukIvan

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Language:PythonApache-2.03083700

botorch

Bayesian optimization in PyTorch

Language:Jupyter NotebookMIT301600

face-alignment

:fire: 2D and 3D Face alignment library build using pytorch

Language:PythonBSD-3-Clause690300

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

Language:PythonApache-2.012954800

gpt-2

Code for the paper "Language Models are Unsupervised Multitask Learners"

Language:PythonNOASSERTION2200700

Smelter

Overhead-free ONNX graph inference engine with Metal under the hood

Language:SwiftMIT5000

light-weight-refinenet

Light-Weight RefineNet for Real-Time Semantic Segmentation

Language:PythonNOASSERTION72700

PySnooper

Never use print for debugging again

Language:PythonMIT1630400

manim

Animation engine for explanatory math videos

Language:PythonMIT6081300

albumentations

Fast and flexible image augmentation library. Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

Language:PythonMIT1378700

baselines

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms

Language:PythonMIT1552800

chroma-key-composition

The objective of this work is to implement python-based chroma-key script to remove green background from an image and merge it to a background.

Language:Jupyter Notebook1800

ml-agents

The Unity Machine Learning Agents Toolkit (ML-Agents) is an open-source project that enables games and simulations to serve as environments for training intelligent agents using deep reinforcement learning and imitation learning.

Language:C#NOASSERTION1667800