Files

pedagogical_examples

Rafi Witten

Mar 19, 2024

2589376 · Mar 19, 2024

This branch is 1 commit ahead of, 1606 commits behind AI-Hypercomputer/maxtext:main.

Name	Name	Last commit message	Last commit date
parent directory ..
non_spmd.py	non_spmd.py	Added a basic not-SPMD example.	Nov 20, 2023
shardings.py	shardings.py	Adds microbenchmarking and uses standard 1e9 for memory calcs.	Mar 16, 2024
shmap_collective_matmul.py	shmap_collective_matmul.py	Added pedagogical example/shamp_collective_matmul.py	Mar 19, 2024