Steven Dalton's starred repositories
cuda-python
CUDA Python Low-level Bindings
adahessian
ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning
pytorch-lightning
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
SPIRAL-tensorflow
in progress
cudnn-training
A CUDNN minimal deep learning training code sample using LeNet.
tensorboardX
tensorboard for pytorch (and chainer, mxnet, numpy, ...)
pytorch-cifar
95.47% on CIFAR10 with PyTorch
alpha-zero-general
A clean implementation based on AlphaZero for any game in any framework + tutorial + Othello/Gobang/TicTacToe/Connect4 and more
line_profiler
(OLD REPO) Line-by-line profiling for Python - Current repo ->
TensorComprehensions
A domain specific language to express machine learning workloads.
pytorch_fft
PyTorch wrapper for FFTs
DCGAN-tensorflow
A tensorflow implementation of "Deep Convolutional Generative Adversarial Networks"