Skip to content
View YanjieZe's full-sized avatar
537 followers · 86 following

Achievements

Achievement: Starstruckx3Achievement: QuickdrawAchievement: Pull Shark

Achievements

Achievement: Starstruckx3Achievement: QuickdrawAchievement: Pull Shark

Highlights

  • Pro

Block or report YanjieZe

Block user

Prevent this user from interacting with your repositories and sending you notifications. Learn more about blocking users.

You must be logged in to block users.

Please don't include any personal information such as legal names or email addresses. Maximum 100 characters, markdown supported. This note will be visible to only you.
Report abuse

Contact GitHub support about this user’s behavior. Learn more about reporting abuse.

Report abuse
Stars

Torch

tools for torch
10 repositories

pytorch / TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Python 2,624 355 Updated Dec 22, 2024

rentruewang / koila

Prevent PyTorch's `CUDA error: out of memory` in just 1 line of code.

Python 1,821 63 Updated Nov 29, 2024

fossasia / visdom

A flexible tool for creating, organizing, and sharing visualizations of live, rich data. Supports Torch and Numpy.

Python 10,059 1,136 Updated May 30, 2024

cupy / cupy

NumPy & SciPy for GPU

Python 9,605 861 Updated Dec 20, 2024

TylerYep / torchinfo

View model summaries in PyTorch!

Python 2,641 124 Updated Dec 17, 2024

luo3300612 / Visualizer

assistant tools for attention visualization in deep learning

Jupyter Notebook 1,058 82 Updated Jun 9, 2022

omry / omegaconf

Flexible Python configuration system. The last one you will ever need.

Python 2,001 116 Updated May 30, 2024

szagoruyko / pytorchviz

A small package to create visualizations of PyTorch execution graphs

Jupyter Notebook 3,256 281 Updated Apr 2, 2024

google / learned_optimization

Python 755 63 Updated Nov 14, 2024

jax-ml / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Python 30,818 2,844 Updated Dec 22, 2024