fuhailin

composer

Supercharge Your Model Training

neural-collaborative-filtering

pytorch version of neural collaborative filtering

CUDA-GEMM-Optimization

CUDA Matrix Multiplication Optimization

glake

GLake: optimizing GPU memory management and IO transmission.

OpenRLHF

An Easy-to-use, Scalable and High-performance RLHF Framework (Support 70B+ full tuning & LoRA & Mixtral & KTO)

tensorflow-nanoGPT

Example how to train GPT-2 (XLA + AMP), export to SavedModel and serve with Tensorflow Serving

flash-attention

Fast and memory-efficient exact attention

monolith

ByteDance's Recommendation System

effective_transformer

Running BERT without Padding

libtensorflow_cc

Pre-built libtensorflow_cc.so and Docker Images for TensorFlow C++ API

dlrover

DLRover: An Automatic Distributed Deep Learning System

xformers

Hackable and optimized Transformers building blocks, supporting a composable construction.

DALI

A GPU-accelerated library containing highly optimized building blocks and an execution engine for data processing to accelerate deep learning training and inference applications.

RecommendFlow

A Industrialization recommendation system which can easy to build pipeline between local/hdfs train data and tensorflow2.x model.

keras-core

A multi-backend implementation of the Keras API, with support for TensorFlow, JAX, and PyTorch.

trellis

General purpose middleware framework for distributed applications

semantic-mediawiki

A repository to continuously release Semantic MediaWiki (SMW) Docker images that track the latest SMW stable updates.

bazel-action

A GitHub action to run @bazelbuild commands

pybind11_bazel

Bazel wrapper around the pybind11 repository

RecLearn

Recommender Learning with Tensorflow2.x

LoRA

Code for loralib, an implementation of "LoRA: Low-Rank Adaptation of Large Language Models"

EasyParallelLibrary

Easy Parallel Library (EPL) is a general and efficient deep learning framework for distributed model training.

dragonfly

A modern replacement for Redis and Memcached

OpenNMT-tf

Neural machine translation and sequence learning using TensorFlow

RecSys_course

Course on recommender systems conducted at the Faculty of Computer Science, National Research University - Higher School of Economics. Academic year 2022-2023.

GradientAccumulator

:dart: Accumulated Gradients for TensorFlow 2

multistage-recommenders

Oh-My-Dockerfile

A collection of dockerfiles for various tasks

backgroundremover

Background Remover lets you Remove Background from images and video using AI with a simple command line interface that is free and open source.

BERT_text_classification

How to fine-tune a BERT classifier for detecting the sentiment of a movie review and the toxicity of a comment.