Pinned Repositories
alpa
Training and serving large-scale neural networks with auto parallelization.
apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
containers
Containers
dgl
Python package built to ease deep learning on graph, on top of existing DL frameworks.
diffusers
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
dotfiles
Dotfiles
ds-elastic-exper
stock_prediction
marwage's Repositories
marwage/stock_prediction
marwage/alpa
Training and serving large-scale neural networks with auto parallelization.
marwage/apex
A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch
marwage/containers
Containers
marwage/dgl
Python package built to ease deep learning on graph, on top of existing DL frameworks.
marwage/diffusers
🤗 Diffusers: State-of-the-art diffusion models for image and audio generation in PyTorch and FLAX.
marwage/dotfiles
Dotfiles
marwage/ds-elastic-exper
marwage/global-batch-norm
marwage/hugo-theme-introduction
Minimal, single page, smooth-scrolling theme for Hugo static site generator.
marwage/marwage
Config files for my GitHub profile.
marwage/mindspore
MindSpore is a new open source deep learning training/inference framework that could be used for mobile, edge and cloud scenarios.
marwage/gnn_memory_management
Loss memory on the GPU
marwage/jax_playground
marwage/marwage.github.io
marwage/models
Models and examples built with TensorFlow
marwage/nccl-test
marwage/philly-traces
marwage/Plot
marwage/pytorch_geometric
Forked to modify the examples
marwage/Scarlett
Scarlett
marwage/Talulla
Talulla
marwage/todo_polish
ToDo Polish