Pinned Repositories
BrainPy
Brain Dynamics Programming in Python
brainpy-largescale
mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap:
jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
brainpy-largescale
Dong-Jiahuan's Repositories
Dong-Jiahuan/BrainPy
Brain Dynamics Programming in Python
Dong-Jiahuan/brainpy-largescale
Dong-Jiahuan/mpi4jax
Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap: