Linking pages
- Supercharged high-resolution ocean simulation with JAX | dionhaefner.github.io https://dionhaefner.github.io/2021/12/supercharged-high-resolution-ocean-simulation-with-jax/ 48 comments
- dionhaefner (Dion Häfner) · GitHub https://github.com/dionhaefner 0 comments
- JAX Ecosystem. JAX by Google Research is getting more… | by Grigory Sapunov | Medium https://moocaholic.medium.com/jax-a13e83f49897 0 comments
Linked pages
- TensorFlow http://tensorflow.org/ 440 comments
- Taichi Lang: High-performance Parallel Programming in Python https://www.taichi-lang.org/ 103 comments
- PyTorch http://pytorch.org/ 100 comments
- GitHub - google/jax: Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more https://github.com/google/jax 99 comments
- Numba: A High Performance Python Compiler https://numba.pydata.org/ 67 comments
- CuPy: NumPy & SciPy for GPU https://cupy.chainer.org/ 33 comments
- NumPy https://numpy.org 17 comments
- GitHub - team-ocean/veros: The versatile ocean simulator, in pure Python, powered by JAX. https://github.com/dionhaefner/veros 9 comments