alpha0422 Goto Github PK

apex

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

attn-score

Higher performance attention score calculation.

backward-cpp

A beautiful stack trace pretty printer for C++

blog-comments

bprop

Reinvent back propagation for common operations to get a better understanding.

conv2d-fft

FFT tiling implementation for strided convolution dgrad.

deepspeech

A TensorFlow implementation of Baidu's DeepSpeech architecture

deepspeech.pytorch

Speech Recognition using DeepSpeech2 and the CTC activation function. Edit

digest

dist-opt

PyTorch distributed optimizer C++ implementation.

dist-optimizer

Distributed optimizer for PyTorch.

focal-loss-cuda

Efficient CUDA focal loss implementation for PyTorch.

hugo-coder

A minimalist blog theme for hugo.

label-smoothing-cuda

High performance implementation of CUDA label smoothing with softmax cross entropy loss.

lightning

Build and train PyTorch models and connect them to the ML lifecycle using Lightning App templates, without handling DIY infrastructure, cost management, scaling, and other headaches.

lstm_perf

LSTM performance study.

megatron-lm

Ongoing research training transformer models at scale

mha

Multi-head attention.

nemo

NeMo: a framework for generative AI

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

slurm-docker-mpi

Run OMPI application inside docker container with SLURM.

stable-diffusion

A latent text-to-image diffusion model

stablediffusion

High-Resolution Image Synthesis with Latent Diffusion Models

study

Study for various topics.

torch-graph

Simple PyTorch graph capturing.

tpu

Reference models and tools for Cloud TPUs.

transformers

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

utils

General utilities.

vnpy

基于Python的开源量化交易平台开发框架