Pinned Repositories
ACProp-Optimizer
Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)
Adabelief-Optimizer
Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"
explain_invertible
repo for "Decision explanation and feature importance for invertible networks"
GSAM
PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)
LadderNet
ShelfNet
implementation for paper "ShelfNet for fast semantic segmentation"
ShelfNet-lw-cityscapes
ShelfNet-lightweight for paper [ShelfNet for fast semantic segmentation]
SNGAN-AdaBelief
Train SN-GAN with AdaBelief
torch_ACA
repo for paper: Adaptive Checkpoint Adjoint (ACA) method for gradient estimation in neural ODE
TorchDiffEqPack
juntang-zhuang's Repositories
juntang-zhuang/Adabelief-Optimizer
Repository for NeurIPS 2020 Spotlight "AdaBelief Optimizer: Adapting stepsizes by the belief in observed gradients"
juntang-zhuang/GSAM
PyTorch repository for ICLR 2022 paper (GSAM) which improves generalization (e.g. +3.8% top-1 accuracy on ImageNet with ViT-B/32)
juntang-zhuang/TorchDiffEqPack
juntang-zhuang/ACProp-Optimizer
Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)
juntang-zhuang/SNGAN-AdaBelief
Train SN-GAN with AdaBelief
juntang-zhuang/machine-learning-notes
My continuously updated Machine Learning, Probabilistic Models and Deep Learning notes and demos (2000+ slides) 我不间断更新的机器学习，概率模型和深度学习的讲义(2000+页)和视频链接
juntang-zhuang/addons
Useful extra functionality for TensorFlow 2.x maintained by SIG-addons
juntang-zhuang/alphafold2
To eventually become an unofficial Pytorch implementation / replication of Alphafold2, as details of the architecture get released
juntang-zhuang/CS-Notes
:books: 技术面试必备基础知识、Leetcode、计算机操作系统、计算机网络、系统设计
juntang-zhuang/datasets
🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools
juntang-zhuang/fairseq
Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
juntang-zhuang/flax
Flax is a neural network library for JAX that is designed for flexibility.
juntang-zhuang/juntang-zhuang.github.io
juntang-zhuang/keras
Deep Learning for humans
juntang-zhuang/mit-deep-learning-book-pdf
MIT Deep Learning Book in PDF format (complete and parts) by Ian Goodfellow, Yoshua Bengio and Aaron Courville
juntang-zhuang/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
juntang-zhuang/PyTorch-Wavelet-Toolbox
Differentiable fast wavelet transforms in PyTorch with GPU support.
juntang-zhuang/symplectic-adjoint-method
Code for "Symplectic Adjoint Method for Exact Gradient of Neural ODE with Minimal Memory," NeurIPS, 2021.
juntang-zhuang/system-design-primer
Learn how to design large-scale systems. Prep for the system design interview. Includes Anki flashcards.
juntang-zhuang/Uni-Fold
juntang-zhuang/alphageometry
juntang-zhuang/awesome-chatgpt-prompts
This repo includes ChatGPT prompt curation to use ChatGPT better.
juntang-zhuang/big_vision
Official codebase used to develop Vision Transformer, MLP-Mixer, LiT and more.
juntang-zhuang/gptstore-prompts
Here are the Top 100 prompts on GPTStore, which we can use to learn and improve prompt engineering.
juntang-zhuang/juntang-zhuang
juntang-zhuang/large-sequence-modeling
Transformers with Arbitrarily Large Context, No Approximations
juntang-zhuang/PyTorch-fpilayer
juntang-zhuang/search_with_lepton
Building a quick conversation-based search demo with Lepton AI.
juntang-zhuang/Trinkle23897
juntang-zhuang/vision_transformer