Skip to content

Latest commit

 

History

History
17 lines (14 loc) · 286 Bytes

README.md

File metadata and controls

17 lines (14 loc) · 286 Bytes

Science_Bacc

SQA scottish Science Baccalaureate of Christopher Rae. Improving the preformance of distributed data parallelism of low wifi bandwidths

models found in train_model.py

Requirements

  • MPI
  • Jax
  • mpi4jax
  • flax
  • optax

RUN

chmod +x ./run.sh
sh ./run.sh