distributed/tensor_parallelism - pytorch/examples

A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.

0 0 45 Python

	fsdp_tp_example.py	6.0 KB
	llama2_model.py	16.2 KB
	log_utils.py	611 B
	README.md	1.1 KB
	requirements.txt	96 B
	run_example.sh	497 B
	sequence_parallel_example.py	3.2 KB
	tensor_parallel_example.py	3.9 KB