Kh4L

DALI

A library containing both highly optimized building blocks and an execution engine for data pre-processing in deep learning applications

flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

gluon-cv

Gluon CV Toolkit

pytorch-i-revnet

i-RevNet Pytorch Code

cocoapi

COCO API - Dataset @ http://cocodataset.org/

DALI_extra

Test data for DALI project

DeepLearningExamples

Deep Learning Examples

dgl

Python package built to ease deep learning on graph, on top of existing DL frameworks.

docs

TensorFlow documentation

efficient-node-labelling

Implementation for Message Passing Link Predictor (MPLP)

gluon-nlp

NLP made easy

horovod

Distributed training framework for TensorFlow, Keras, PyTorch, and Apache MXNet.

incubator-mxnet

Lightweight, Portable, Flexible Distributed/Mobile Deep Learning with Dynamic, Mutation-aware Dataflow Dep Scheduler; for Python, R, Julia, Scala, Go, Javascript and more

kaolin

A PyTorch Library for Accelerating 3D Deep Learning Research

linuxconf

My Linux (Debian/Ubuntu) base conf

MMGL

Multimodal Graph Learning: how to encode multiple multimodal neighbors with their relations into LLMs

ogb

Benchmark datasets, data loaders, and evaluators for graph machine learning

onnx

Open standard for machine learning interoperability

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-distributed-example

pytorch-transformers

👾 A library of state-of-the-art pretrained models for Natural Language Processing (NLP)

pytorch_geometric

Graph Neural Network Library for PyTorch

relbench

RelBench: Relational Deep Learning Benchmark

stable_diffusion_playground

Playing around with stable diffusion. Generated images are reproducible because I save the metadata and latent information. You can generate and then later interpolate between the images of your choice.

stanford_alpaca

Code and documentation to train Stanford's Alpaca models, and generate the data.

TAPE

Official Implementation of ICLR 2024 paper "Harnessing Explanations: LLM-to-LM Interpreter for Enhanced Text-Attributed Graph Representation Learning"

tensorflow

An Open Source Machine Learning Framework for Everyone

TensorRT

TensorRT is a C++ library for high performance inference on NVIDIA GPUs and deep learning accelerators.

tensorrt-tf

TensorFlow/TensorRT integration

tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators