ProJAX

This is project that reproduce papers in JAX.

Name		Name	Last commit message	Last commit date
Latest commit History 37 Commits
.vscode		.vscode
ChatGPT_tasks		ChatGPT_tasks
Flax		Flax
Jax		Jax
NeuralODE		NeuralODE
docker		docker
haiku_examples		haiku_examples
tensorflow-metal		tensorflow-metal
.DS_Store		.DS_Store
.flake8		.flake8
.gitignore		.gitignore
Distributed_arrays_and_automatic_parallelization.ipynb		Distributed_arrays_and_automatic_parallelization.ipynb
Introduction_to_sharded_computation.ipynb		Introduction_to_sharded_computation.ipynb
LICENSE		LICENSE
README.md		README.md
requirements.txt		requirements.txt

Provide feedback