jglaser

affinity_docker

Docker image for affinity prediction

AutoDock-Vina

AutoDock Vina

cupy

NumPy & SciPy for GPU

dask-cuda

Utilities for Dask and CUDA interactions

dask-mpi

Deploy Dask using MPI4Py

datasets

🤗 The largest hub of ready-to-use NLP datasets for ML models with fast, easy-to-use and efficient data manipulation tools

DeepSpeed

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

DeepSpeedExamples

Example models using DeepSpeed

gigadocking_notebooks

Collect colab notebooks for Gigadocking analysis

gpu-bdb

RAPIDS GPU-BDB

groqflow

GroqFlow provides an automated tool flow for compiling machine learning and linear algebra workloads into Groq programs and executing those programs on GroqChip™ processors.

h-transformer-1d

Implementation of H-Transformer-1D, Hierarchical Attention for Sequence Learning

HIP

HIP : Convert CUDA to Portable C++ Code

hoomd-blue

Molecular dynamics and Monte Carlo soft matter simulation on GPUs.

linear_mem_attention_pytorch

Unofficially Implements https://arxiv.org/abs/2112.05682 to get Linear Memory Cost on Attention for PyTorch

NeMo

NeMo: a toolkit for conversational AI

neural-tangents

Fast and Easy Infinite Neural Networks in Python

onnxconverter-common

Common utilities for ONNX converters

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-optimizer

torch-optimizer -- collection of optimizers for Pytorch

random123

Tracking DE Shaw Research's random123 library

rocm_smi_lib

ROCm SMI LIB

sentence-transformers

Multilingual Sentence & Image Embeddings with BERT

tensor-cores-numerical-behavior

Test suite for probing the numerical behavior of NVIDIA tensor cores

tensorflow-upstream

TensorFlow ROCm port

transformers

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

tsl

TwoFold_DL

Deep Learning model for protein and ligand complex structure prediction from sequences and SMILES

wandb

🔥 A tool for visualizing and tracking your machine learning experiments. This repo contains the CLI and Python API.

xla

A machine learning compiler for GPUs, CPUs, and ML accelerators