├── .gitignore ├── LICENSE ├── README.md ├── euler.py ├── euler_distributed.py ├── euler_numpy.py ├── make_gif.sh ├── requirements.txt ├── results ├── plot_results.py ├── result_16384_single.png ├── runs.json ├── scaling_strong.png └── scaling_weak.png ├── sbatch_profile.sh └── sbatch_rusty_gpu.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/README.md -------------------------------------------------------------------------------- /euler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/euler.py -------------------------------------------------------------------------------- /euler_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/euler_distributed.py -------------------------------------------------------------------------------- /euler_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/euler_numpy.py -------------------------------------------------------------------------------- /make_gif.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/make_gif.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | jax[cuda12] 2 | numpy 3 | -------------------------------------------------------------------------------- /results/plot_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/results/plot_results.py -------------------------------------------------------------------------------- /results/result_16384_single.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/results/result_16384_single.png -------------------------------------------------------------------------------- /results/runs.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/results/runs.json -------------------------------------------------------------------------------- /results/scaling_strong.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/results/scaling_strong.png -------------------------------------------------------------------------------- /results/scaling_weak.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/results/scaling_weak.png -------------------------------------------------------------------------------- /sbatch_profile.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/sbatch_profile.sh -------------------------------------------------------------------------------- /sbatch_rusty_gpu.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/pmocz/jax-euler-benchmarks/HEAD/sbatch_rusty_gpu.sh --------------------------------------------------------------------------------