sbhavani / jax-aws-docker Public

Notifications You must be signed in to change notification settings
Fork 0
Star 0

Example Docker image for JAX on AWS with EFA

0 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 20 Commits
aws-pyt		aws-pyt
jax-toolbox		jax-toolbox
ngc		ngc
pyt-test-comms		pyt-test-comms
README.md		README.md
jax_paxml.Dockerfile.orig		jax_paxml.Dockerfile.orig
train.py		train.py

Repository files navigation

jax-aws-docker

Example Docker image for JAX on AWS with EFA based on https://github.com/aws-samples/awsome-distributed-training/tree/main/3.test_cases/jax

About

Example Docker image for JAX on AWS with EFA

Report repository

Releases

No releases published

Packages

No packages published

Languages