bmedishe Goto Github PK

apex

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

bitsandbytes

Accessible large language models via k-bit quantization for PyTorch.

bmedishe

Config files for my GitHub profile.

build

Boost.Build

deepspeed

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

diffusers

🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch

fambench

Benchmarks to capture important workloads.

fbgemm

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

gloo

Collective communications library with various primitives for multi-machine training.

hipify_torch

huggingface-transformers

🤗Transformers: State-of-the-art Natural Language Processing for Pytorch and TensorFlow 2.0.

kair

Image Restoration Toolbox (PyTorch). Training and testing codes for DPIR, USRNet, DnCNN, FFDNet, SRMD, DPSR, BSRGAN, SwinIR

onnxruntime

ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

torchprof

PyTorch layer-by-layer model profiler

torchrec

Pytorch domain library for recommendation systems

transformers-deepspeed

vision

Datasets, Transforms and Models specific to Computer Vision