Popular repositories Loading

1. torch-graph torch-graph Public

   Simple PyTorch graph capturing.

   Python 21 2

2. label-smoothing-cuda label-smoothing-cuda Public archive

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

   Cuda 6 3

3. deepspeech.pytorch deepspeech.pytorch Public

   Forked from SeanNaren/deepspeech.pytorch

   Speech Recognition using DeepSpeech2 and the CTC activation function. Edit

   Python 1

4. focal-loss-cuda focal-loss-cuda Public

   Efficient CUDA focal loss implementation for PyTorch.

   Cuda 1

5. TransformerEngine TransformerEngine Public

   Forked from NVIDIA/TransformerEngine

   A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilizatio…

   Python 1

6. pytorch pytorch Public

   Forked from pytorch/pytorch

   Tensors and Dynamic neural networks in Python with strong GPU acceleration

   Python