Pinned Repositories
alpa
Training and serving large-scale neural networks with auto parallelization.
ParlAI
A framework for training and evaluating AI models on a variety of openly available dialogue datasets.
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
alpa
Auto parallelization for large-scale neural networks
Learn2Smile
Learn2Smile: Learning non-verbal interaction through observation. Accepted by IROS 2017: https://ieeexplore.ieee.org/document/8206272
sccache
sccache with CUDA compilation (NVCC) support
yf225's Repositories
yf225/Learn2Smile
Learn2Smile: Learning non-verbal interaction through observation. Accepted by IROS 2017: https://ieeexplore.ieee.org/document/8206272
yf225/alpa
Auto parallelization for large-scale neural networks
yf225/ao
The missing pytorch dtype and layout library for training and inference
yf225/benchmark
yf225/composable_kernel
yf225/fairscale
PyTorch extensions for high performance and large scale training.
yf225/functorch
functorch is JAX-like composable function transforms for PyTorch.
yf225/intrinsic3d
Intrinsic3D - High-Quality 3D Reconstruction by Joint Appearance and Geometry Optimization with Spatially-Varying Lighting (ICCV 2017)
yf225/jax-alpa
yf225/lightseq
LightSeq: A High Performance Library for Sequence Processing and Generation
yf225/llvm-project
The LLVM Project is a collection of modular and reusable compiler and toolchain technologies. Note: the repository does not accept github pull requests at this moment. Please submit your patches at http://reviews.llvm.org.
yf225/Megatron-LM
Ongoing research training transformer language models at scale, including: BERT & GPT-2
yf225/Meshroom
3D Reconstruction Software
yf225/metaseq
Repo for external large-scale work
yf225/nvdiffrec
Official code for the CVPR 2022 (oral) paper "Extracting Triangular 3D Models, Materials, and Lighting From Images".
yf225/pip-openexr
yf225/protoquant
Prototype routines for GPU quantization written using PyTorch.
yf225/pycls
Codebase for Image Classification Research, written in PyTorch.
yf225/pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
yf225/pytorch-image-models
PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more
yf225/rocBLAS
Next generation BLAS implementation for ROCm platform
yf225/tensorflow-alpa
yf225/torchdistx
Torch Distributed Experimental
yf225/torchdynamo
A Python-level JIT compiler designed to make unmodified PyTorch programs faster.
yf225/torchoptim
yf225/torchtune
A Native-PyTorch Library for LLM Fine-tuning
yf225/triton
Development repository for the Triton language and compiler
yf225/use-ray-with-slurm
A brief tutorial on how to use ray/rllib/tun in slurm cluster.
yf225/vision_transformer
yf225/vit-tf