Zhiyuan Chen's repositories
Codes-for-Lane-Detection
Tensorflow implementation of "Spatial As Deep: Spatial CNN for Traffic Scene Understanding" (AAAI 2018)
PyContrast
PyTorch implementation of Contrastive Learning methods; List of awesome-contrastive-learning papers
accelerate
🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
basex
Fast Base X encoder/decoder
DeepSpeed
DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.
flash-attention
Fast and memory-efficient exact attention
huggingface_hub
All the open source things related to the Hugging Face Hub.
mkdocs-material
Documentation that simply works
mkdocs-material-extensions-2
Extension pack for Python Markdown and MkDocs Material.
mkdocs-rss-plugin
MkDocs plugin to generate a RSS feeds for created and updated pages, using git log and YAML frontmatter (page.meta).
MultiRM
Code for paper 'Attention-based multi-label neural networks for integrated prediction and interpretation of twelve widely occurring RNA modifications'
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
setuptools_scm
the blessed package to manage your versions by scm tags
torcheval
A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.
transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
visual-chatgpt
Official repo for the paper: Visual ChatGPT: Talking, Drawing and Editing with Visual Foundation Models