https://github.com/graphcore-research/jax-experimental
JAX for Graphcore IPU (experimental)
https://github.com/graphcore-research/jax-experimental
ai deep-learning graphcore ipu jax machine-learning
Last synced: 5 months ago
JSON representation
JAX for Graphcore IPU (experimental)
- Host: GitHub
- URL: https://github.com/graphcore-research/jax-experimental
- Owner: graphcore-research
- License: apache-2.0
- Created: 2023-01-20T15:50:14.000Z (over 3 years ago)
- Default Branch: jax-v0.3.16-ipu
- Last Pushed: 2024-03-12T11:03:46.000Z (over 2 years ago)
- Last Synced: 2025-07-27T06:01:08.214Z (11 months ago)
- Topics: ai, deep-learning, graphcore, ipu, jax, machine-learning
- Language: Python
- Homepage: https://github.com/graphcore-research/jax-experimental#readme
- Size: 49.7 MB
- Stars: 21
- Watchers: 5
- Forks: 2
- Open Issues: 5