Awesome Open Source

Programming Languages

Search results for pytorch nvidia

126 search results found

Deeplearningexamples ⭐ 12,073

State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.

Fastphotostyle ⭐ 11,014

Style transfer, deep learning, feature transform

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

Torch2trt ⭐ 4,267

An easy to use PyTorch to TensorRT converter

Kaolin ⭐ 4,038

A PyTorch Library for Accelerating 3D Deep Learning Research

Alpha Zero General ⭐ 3,497

A clean implementation based on AlphaZero for any game in any framework + tutorial + Othello/Gobang/TicTacToe/Connect4 and more

Tensorrt ⭐ 2,308

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Qualityscaler ⭐ 1,519

QualityScaler - image/video AI upscaler app

Jetson Containers ⭐ 1,388

Machine Learning Containers for NVIDIA Jetson and JetPack-L4T

Densedepth ⭐ 1,365

High Quality Monocular Depth Estimation via Transfer Learning

Dl Colab Notebooks ⭐ 1,314

Try out deep learning models online on Google Colab

Shark ⭐ 1,292

SHARK - High Performance Machine Learning Distribution

Flowframes ⭐ 1,250

Flowframes Windows GUI for video interpolation using DAIN (NCNN) or RIFE (CUDA/NCNN)

Pytorch Qrnn ⭐ 1,235

PyTorch implementation of the Quasi-Recurrent Neural Network - up to 16 times faster than NVIDIA's cuDNN LSTM

Deepstream Yolo ⭐ 1,184

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 / 5.1 implementation for YOLO models

Docker Pytorch ⭐ 872

A Docker image for PyTorch

Redisai ⭐ 799

A Redis module for serving tensors and executing deep learning graphs

Trt_pose ⭐ 763

Real-time pose estimation accelerated with NVIDIA TensorRT

Retinanet Examples ⭐ 727

Fast and accurate object detection with end-to-end GPU optimization

FSGAN - Official PyTorch Implementation

Openchem ⭐ 450

OpenChem: Deep Learning toolkit for Computational Chemistry and Drug Design Research

All-in-one AI container for rapid prototyping

Deeplabv3 ⭐ 388

PyTorch implementation of DeepLabV3, trained on the Cityscapes dataset.

Pytorch Correlation Extension ⭐ 385

Custom implementation of Corrleation Module

MelGAN vocoder (compatible with NVIDIA/tacotron2)

02456 Deep Learning With Pytorch ⭐ 262

Exercises and supplementary material for the deep learning course 02456 using PyTorch.

Onnxruntime Training Examples ⭐ 251

Examples for using ONNX Runtime for model training.

Pipedream ⭐ 239

Pytorch Docker ⭐ 208

Pure Pytorch Docker Images.

ESANet: Efficient RGB-D Semantic Segmentation for Indoor Scene Analysis

Realscaler ⭐ 199

RealScaler - image/video AI upscaler app (Real-ESRGAN)

Pytorch Aarch64 ⭐ 191

PyTorch wheels (whl) & conda for aarch64 / ARMv8 / ARM64

Onnx Web ⭐ 156

web UI for GPU-accelerated ONNX pipelines like Stable Diffusion, even on Windows and AMD

Autoclint ⭐ 150

A specially designed light version of Fast AutoAugment

Frontalization ⭐ 139

Pytorch deep learning face frontalization model

Ds_bowl_2018 ⭐ 112

Kaggle Data Science Bowl 2018

Deep Smoke Machine ⭐ 110

Deep learning models and dataset for recognizing industrial smoke emissions

Learningrigidity ⭐ 110

Learning Rigidity in Dynamic Scenes with a Moving Camera for 3D Motion Field Estimation (ECCV 2018)

Autocarjetsonnano ⭐ 106

PyTorch Python Neural Network Autonomous 1/10 Car for Nvidia Jetson Nano

Dali_pytorch_demo ⭐ 102

Example code showing how to use Nvidia DALI in pytorch, with fallback to torchvision. Contains a few differences to the official Nvidia example, namely a completely CPU pipeline & improved memory usage

Finbert ⭐ 101

BERT for Finance : UC Berkeley MIDS w266 Final Project

Deepstream Yolo Pose ⭐ 93

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 application for YOLO-Pose models

Pytorch_gbw_lm ⭐ 90

PyTorch Language Model for 1-Billion Word (LM1B / GBW) Dataset

Deep Homography Estimation Pytorch ⭐ 89

Deep homography network with Pytorch

Arch Data Science ⭐ 83

Archlinux PKGBUILDs for Data Science, Machine Learning, Deep Learning, NLP and Computer Vision

Pytorch Poetry Generation ⭐ 79

Detection of Accounting Anomalies using Deep Autoencoder Neural Networks - A lab we prepared for NVIDIA's GPU Technology Conference 2018 that will walk you through the detection of accounting anomalies using deep autoencoder neural networks. The majority of the lab content is based on Jupyter Notebook, Python and PyTorch.

PyTorch implementation of the ACCV 2018-AIU2018 paper Video Summarization with Attention

Pytorch implementation of our paper accepted by NeurIPS 2020 -- Rotated Binary Neural Network

Progressive Gan Pytorch ⭐ 58

A pytorch implementation of Progressive-GAN that is actually works, readable and simple to customize

Official implementation for our CVPR19 paper, AOGNets: Compositional Grammatical Architectures for Deep Learning

Stable Diffusion Tritonserver ⭐ 49

Deploy stable diffusion model with onnx/tenorrt + tritonserver

A pytorch implementation of "Domain-Adaptive Few-Shot Learning"

Large scale training of factorization models for Collaborative Filtering with PyTorch

This repository contains a docker image that I use to develop my artificial intelligence applications in an uncomplicated fashion. Python, TensorFlow, PyTorch, ONNX, Keras, OpenCV, TensorRT, Numpy, Jupyter notebook... 🐋🔥

Deep_docker ⭐ 43

Setup Deeplearning Tools ⭐ 41

Set up CI in DL/ cuda/ cudnn/ TensorRT/ onnx2trt/ onnxruntime/ onnxsim/ Pytorch/ Triton-Inference-Server/ Bazel/ Tesseract/ PaddleOCR/ NVIDIA-docker/ minIO/ Supervisord on AGX or PC from scratch.

Moldqn Pytorch ⭐ 40

A PyTorch Implementation of "Optimization of Molecules via Deep Reinforcement Learning".

Human Pose Estimation ⭐ 36

simple and lightweight humanpose estimation

Pytorch_integrated_cell ⭐ 36

Integrated Cell project implemented in pytorch

Constant Memory Waveglow ⭐ 36

PyTorch implementation of NVIDIA WaveGlow with constant memory cost.

Turnkeyml ⭐ 34

The AI insights toolchain

Self-driving AI toy car 🤖🚗.

Tailcalibx ⭐ 32

Pytorch implementation of Feature Generation for Long-Tail Classification by Rahul Vigneswaran, Marc T Law, Vineeth N Balasubramaniam and Makarand Tapaswi

Notebooks ⭐ 30

A docker-based starter kit for machine learning via jupyter notebooks. Designed for those who just want a runtime environment and get on with machine learning. Docker tags:

Recipe Personalization ⭐ 29

EMNLP 2019: Generating Personalized Recipes from Historical User Preferences

Ubuntu For Robotics ⭐ 27

This repository is for setting-up cuda-9/8, nvidia-396/387/384 driver, OpenCV-3.3, ROS Kinetic, Tensorflow-1.11/1.7/1.4/1.2.1, Pytorch-0.4

Swish Torch ⭐ 23

Swish Activation - PyTorch CUDA Implementation

Pytorch Mask Rcnn Samples ⭐ 22

Example notebooks on building PyTorch, preparing data and training as well as an updated project from a PyTorch MaskRCNN port

Tensorrt Torch2trt ⭐ 22

A guide for TensorRT and Torch2TRT

Distributed Pytorch ⭐ 22

Distributed, mixed-precision training with PyTorch

Nvidia_tx_utils ⭐ 22

Scripts and installation tips for Jetson TX1 and TX2 embedded GPU devices. Scripts for installing Tensorflow, PyTorch, ROS etc.

Wage.pytorch ⭐ 21

Reproduction of WAGE in PyTorch.

Pytorch Rpi ⭐ 20

Share PyTorch binaries built for Raspberry Pi

Gpu Jupyterhub ⭐ 20

Setting up a Jupyterhub Dockercontainer to spawn Jupyter Notebooks with GPU support (containing Tensorflow, Pytorch and Keras)

PyTorch implementation of "Jasper: An End-to-End Convolutional Neural Acoustic Model" (INTERSPEECH 2019)

Deep Learning ⭐ 19

It contains the coursework and the practice I have done while learning Deep Learning.🚀 👨‍💻💥 🚩🌈

Nixos Nvidia Cuda Python Docker Compose ⭐ 18

A step-by-step guide to setting up Nvidia GPUs with CUDA support running on Docker (and Compose) containers on NixOS host

Code base for SRSGD

Pytorch_gpu_memory ⭐ 17

pytorch gpu memory check

Aime Ml Containers ⭐ 16

AIME machine learning container management system

Dlhardware ⭐ 15

Overview of hardware for deep learning

Drake Torch ⭐ 14

Example demonstrating pytorch c++ integration with drake

Machinelearning_exercises_python_pytorch ⭐ 14

機械学習アルゴリズムの PyTorch 実装の練習コード集。

Nvidia_xavier_utils ⭐ 14

Scripts and installation tips for Jetson AGX Xavier embedded GPU device. Scripts for installing Tensorflow, PyTorch, ROS etc.

Anaconda3 ⭐ 14

Anaconda3, Jupyter Notebook, OpenCV3, TensorFlow and Keras2 for Deep Learning🐳

Deepstream Yolo Seg ⭐ 13

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 implementation for YOLO-Segmentation models

Stylegan2.pytorch ⭐ 12

Face Adversarial Attack ⭐ 12

An easy approach for the competition "Facial Adversary Examples" in TIANCHI

Pytorch implementation of LocaNMF

Nvidia Gpu Ml Library Test ⭐ 11

Simple tests for JAX, PyTorch, and TensorFlow to test if the installed NVIDIA drivers are being properly picked up

Torchfusion Utils ⭐ 11

A pytorch helper library for Mixed Precision Training, Initialization, Metrics and More Utilities to simplify training of deep learning models

Bert Pytorch ⭐ 10

BERT for Distributed PyTorch + AMP Training

Triton Torchgeo Gat Example ⭐ 10

An example of running a pytorch-geometric graph attention model in nvidia triton.

Humpback Whale Identification ⭐ 10

Kaggle Humpback whale identification: 2xGPU Data augmentation + FP16 mixed precision training

Docker Aitools ⭐ 9

Jupyter Lab for local AI work with Pytorch, Tensorflow, OpenCV, OpenAI Gym (based on cudagl 9 image)

Nvidia Accelerated Pytorch Ffmpeg Opencv ⭐ 9

Hardware Accelerated Pytorch Container with (also accelerated) ffmpeg & OpenCV 4

Deepstream Yolo Face ⭐ 9

NVIDIA DeepStream SDK 6.3 / 6.2 / 6.1.1 / 6.1 / 6.0.1 / 6.0 application for YOLO-Face models

Inequity Release ⭐ 8

Code to accompany 2019 paper.

A PyTorch and Fastai based implementation of Self-Organizing Maps.

Related Searches

Python Pytorch (15,943)

Deep Learning Pytorch (7,533)

Jupyter Notebook Pytorch (4,892)

Machine Learning Pytorch (2,934)

Dataset Pytorch (1,848)

Pytorch Convolutional Neural Networks (1,777)

Pytorch Neural Network (1,631)

Pytorch Natural Language Processing (1,408)

Pytorch Computer Vision (1,230)

Pytorch Neural (1,217)

1-100 of 126 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2024 Awesome Open Source. All rights reserved.