GitHub - Mddct/maxtext: DH long context env in jax

TODO:

Name		Name	Last commit message	Last commit date
Latest commit History 1,710 Commits
.github		.github
.vscode		.vscode
MaxText		MaxText
assets		assets
benchmarks		benchmarks
end_to_end		end_to_end
getting_started		getting_started
pedagogical_examples		pedagogical_examples
.gitignore		.gitignore
.pre-commit-config.yaml		.pre-commit-config.yaml
AUTHORS		AUTHORS
CONTRIBUTING.md		CONTRIBUTING.md
LICENSE		LICENSE
PREFLIGHT.md		PREFLIGHT.md
README.md		README.md
code_style.sh		code_style.sh
constraints_gpu.txt		constraints_gpu.txt
docker_build_dependency_image.sh		docker_build_dependency_image.sh
docker_upload_runner.sh		docker_upload_runner.sh
download_dataset.sh		download_dataset.sh
gpu_multi_process_run.sh		gpu_multi_process_run.sh
maxtext_dependencies.Dockerfile		maxtext_dependencies.Dockerfile
maxtext_gpu_dependencies.Dockerfile		maxtext_gpu_dependencies.Dockerfile
maxtext_jax_stable_stack.Dockerfile		maxtext_jax_stable_stack.Dockerfile
maxtext_libtpu_path.Dockerfile		maxtext_libtpu_path.Dockerfile
maxtext_runner.Dockerfile		maxtext_runner.Dockerfile
maxtext_transformerengine_builder.Dockerfile		maxtext_transformerengine_builder.Dockerfile
multihost_job.py		multihost_job.py
multihost_runner.py		multihost_runner.py
preflight.sh		preflight.sh
pylintrc		pylintrc
pyproject.toml		pyproject.toml
requirements.txt		requirements.txt
requirements_with_jax_stable_stack.txt		requirements_with_jax_stable_stack.txt
rto_setup.sh		rto_setup.sh
setup.sh		setup.sh
setup_gcsfuse.sh		setup_gcsfuse.sh
setup_with_retries.sh		setup_with_retries.sh
unit_test_and_lint.sh		unit_test_and_lint.sh