wenwei202

caffe

Caffe for Sparse and Low-rank Deep Neural Networks

terngrad

Ternary Gradients to Reduce Communication in Distributed Deep Learning (TensorFlow)

iss-rnns

Sparse Recurrent Neural Networks -- Pruning Connections and Hidden Sizes (TensorFlow)

autogrow

AutoGrow: Automatic Layer Growing in Deep Convolutional Networks (KDD 2020)

smoothout

SmoothOut: Smoothing Out Sharp Minima to Improve Generalization in Deep Learning

pytorch-examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

FastFlexANN

FastFlexANN is a fast C or C++ language implementation of sparse artificial neural networks with flexible connectivity.

Caffe-Net-Generator

Net prototxt generator for Caffe

algorithms

Implementations of basic data structures and algorithms

benchmarking

Misc benchmarking tools

models

Models built with TensorFlow

loss-landscape

Code for visualizing the loss landscape of neural nets

bi-att-flow

Bidirectional Attention Flow

bigBatch

Code used to generate the results appearing in "Train longer, generalize better: closing the generalization gap in large batch training of neural networks"

caffe2.github.io

The caffe2 website.

CARLsim3

CARLsim is an efficient, easy-to-use, GPU-accelerated software framework for simulating large-scale spiking neural network (SNN) models with a high degree of biological detail.

large-batch-training

Code to reproduce some of the figures in the paper "On Large-Batch Training for Deep Learning: Generalization Gap and Sharp Minima"

mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more

nasbench

NASBench: A Neural Architecture Search Dataset and Benchmark

network-for-relu

nnan

opencv

Open Source Computer Vision Library

opencv_contrib

Repository for OpenCV's extra modules

python-drums

plays drum tablature

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

RecurrentHighwayNetworks

Recurrent Highway Networks - Implementations for Tensorflow, Torch7, Theano and Brainstorm

wenwei202.github.io