ptrblck's starred repositories

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNet-V3/V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

Language:PythonLicense:Apache-2.0Stargazers:29681Issues:304Issues:863

folly

An open-source C++ library developed and used at Facebook.

Language:C++License:Apache-2.0Stargazers:27043Issues:1021Issues:1100

numpy

The fundamental package for scientific computing with Python.

Language:PythonLicense:NOASSERTIONStargazers:26298Issues:593Issues:12348

onnx

Open standard for machine learning interoperability

Language:PythonLicense:Apache-2.0Stargazers:16797Issues:437Issues:2687

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

Language:PythonLicense:Apache-2.0Stargazers:12984Issues:168Issues:2658

maskrcnn-benchmark

Fast, modular reference implementation of Instance Segmentation and Object Detection algorithms in PyTorch.

Language:PythonLicense:MITStargazers:9244Issues:178Issues:1063

vid2vid

Pytorch implementation of our method for high-resolution (e.g. 2048x1024) photorealistic video-to-video translation.

Language:PythonLicense:NOASSERTIONStargazers:8495Issues:247Issues:166

apex

A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

Language:PythonLicense:BSD-3-ClauseStargazers:8018Issues:99Issues:1151

NYU-DLSP20

NYU Deep Learning Spring 2020

Language:Jupyter NotebookLicense:NOASSERTIONStargazers:6619Issues:136Issues:88

deep-learning-v2-pytorch

Projects and exercises for the latest Deep Learning ND program https://www.udacity.com/course/deep-learning-nanodegree--nd101

Language:Jupyter NotebookLicense:MITStargazers:5165Issues:175Issues:155

DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

Language:C++License:Apache-2.0Stargazers:4906Issues:92Issues:1545

tensorwatch

Debugging, monitoring and visualization for Python Machine Learning and Data Science

Language:Jupyter NotebookLicense:MITStargazers:3368Issues:96Issues:69

XLM

PyTorch original implementation of Cross-lingual Language Model Pretraining.

Language:PythonLicense:NOASSERTIONStargazers:2853Issues:56Issues:334

nogil

Multithreaded Python without the GIL

Language:PythonLicense:NOASSERTIONStargazers:2852Issues:62Issues:118

loss-landscape

Code for visualizing the loss landscape of neural nets

Language:PythonLicense:MITStargazers:2648Issues:32Issues:40

pytorch-kaldi

pytorch-kaldi is a project for developing state-of-the-art DNN/RNN hybrid speech recognition systems. The DNN part is managed by pytorch, while feature extraction, label computation, and decoding are performed with the kaldi toolkit.

python-nubia

A command-line and interactive shell framework.

Language:PythonLicense:NOASSERTIONStargazers:1593Issues:44Issues:46

torchstat

Model analyzer in PyTorch

Language:PythonLicense:MITStargazers:1439Issues:13Issues:44

pytorch-ssd

MobileNetV1, MobileNetV2, VGG based SSD/SSD-lite implementation in Pytorch 1.0 / Pytorch 0.4. Out-of-box support for retraining on Open Images dataset. ONNX and Caffe2 support. Experiment Ideas like CoordConv.

Language:PythonLicense:MITStargazers:1375Issues:39Issues:169

sklearn-porter

Transpile trained scikit-learn estimators to C, Java, JavaScript and others.

Language:PythonLicense:BSD-3-ClauseStargazers:1268Issues:33Issues:70

lightning-thunder

Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enables using different hardware executors at once; across one or thousands of GPUs.

Language:PythonLicense:Apache-2.0Stargazers:987Issues:35Issues:51

pytorch-flows

PyTorch implementations of algorithms for density estimation

Language:PythonLicense:MITStargazers:567Issues:19Issues:8

sparse-to-dense.pytorch

ICRA 2018 "Sparse-to-Dense: Depth Prediction from Sparse Depth Samples and a Single Image" (PyTorch Implementation)

magnet

Deep Learning Projects that Build Themselves

Language:Jupyter NotebookLicense:MITStargazers:359Issues:11Issues:5

mirror

Visualisation tool for CNNs in pytorch

Language:Jupyter NotebookLicense:MITStargazers:250Issues:11Issues:5

Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")

Language:C++License:NOASSERTIONStargazers:225Issues:17Issues:465

Synthetic2Realistic

[ECCV 2018]: T2Net: Synthetic-to-Realistic Translation for Depth Estimation Tasks

DrummerNet

Supplementary material of "Deep Unsupervised Drum Transcription", ISMIR 2019

AICamera

Demonstration of using Caffe2 inside an Android application.

Language:C++License:NOASSERTIONStargazers:67Issues:4Issues:0

JointVAE_v1

Playing around with JointVAE implementation - WIP

Language:Jupyter NotebookStargazers:6Issues:2Issues:0