GitHub - jaglinux/pytorch-notes

Pytorch useful code and examples

Name		Name	Last commit message	Last commit date
Latest commit History 46 Commits
cmake/first_project		cmake/first_project
data_parallel		data_parallel
hip		hip
mx		mx
n-d-list		n-d-list
patch		patch
torch_moe		torch_moe
README.md		README.md
amax.py		amax.py
cross_entropy_loss.py		cross_entropy_loss.py
cuda_CUDA_VISIBLE_DEVICES.py		cuda_CUDA_VISIBLE_DEVICES.py
cuda_device.py		cuda_device.py
cuda_device_count.py		cuda_device_count.py
elastic_ddp.py		elastic_ddp.py
fp32vsfp16.py		fp32vsfp16.py
int8_mm.py		int8_mm.py
matmul.py		matmul.py
maxpool1d.py		maxpool1d.py
maxpool2d.py		maxpool2d.py
memory_stats.py		memory_stats.py
rnn_dropout.py		rnn_dropout.py
softmax.py		softmax.py
tensor_memory_1.py		tensor_memory_1.py
tensor_memory_2.py		tensor_memory_2.py
tf32.py		tf32.py
tf32_perf.py		tf32_perf.py
torch_profiler.py		torch_profiler.py

Provide feedback