pytorch

Mars is a tensor-based unified framework for large-scale data computation which scales numpy, pandas, scikit-learn and Python functions.

Pytorch framework for doing deep learning on point clouds.

Build a modern LLM from scratch. Every line commented. Explained like we are five.

MLNLP：本仓库整理人工智能会议（如 ACL、EMNLP、NAACL、COLING、AAAI、IJCAI、ICLR、NeurIPS、ICML 等）中开源代码的论文。

Resource scheduling and cluster management for AI

My implementation of the original GAT paper (Veličković et al.). I've additionally included the playground.py file for visualizing the Cora dataset, GAT embeddings, an attention mechanism, and entropy histograms. I've supported both Cora (transductive) and PPI (inductive) examples!

Repository for benchmarking graph neural networks (JMLR 2023)

Image Deblurring using Generative Adversarial Networks

Automatically remove the mosaics in images and videos, or add mosaics to them.

EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212

Pytorch domain library for recommendation systems

A PyTorch Library for Multi-Task Learning

The PyTorch-based audio source separation toolkit for researchers

Deep Learning Server and CLI for Torch and TensorRT

Automatic architecture search and hyperparameter optimization for PyTorch

mPLUG-Owl: The Powerful Multi-modal Large Language Model Family

Decentralized deep learning in PyTorch. Built to train models on thousands of volunteers across the world.

[AAAI-23 Oral] Official implementation of the paper "Are Transformers Effective for Time Series Forecasting?"

2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.

Jupyter Notebook tutorials on solving real-world problems with Machine Learning & Deep Learning using PyTorch. Topics: Face detection with Detectron 2, Time Series anomaly detection with LSTM Autoencoders, Object Detection with YOLO v5, Build your first Neural Network, Time Series forecasting for Coronavirus daily cases, Sentiment Analysis with BER