https://github.com/ayaka14732/bart-jax
JAX implementation of BART, aiming to demonstrate how Transformer-based models can be implemented using JAX and trained on Google Cloud TPUs
https://github.com/ayaka14732/bart-jax
bart jax language-model natural-language-processing nlp tpu trans transformer
Last synced: 3 months ago
JSON representation
JAX implementation of BART, aiming to demonstrate how Transformer-based models can be implemented using JAX and trained on Google Cloud TPUs
- Host: GitHub
- URL: https://github.com/ayaka14732/bart-jax
- Owner: ayaka14732
- Created: 2022-11-15T02:05:05.000Z (over 2 years ago)
- Default Branch: main
- Last Pushed: 2022-11-24T14:41:47.000Z (over 2 years ago)
- Last Synced: 2023-03-04T20:03:13.122Z (over 2 years ago)
- Topics: bart, jax, language-model, natural-language-processing, nlp, tpu, trans, transformer
- Language: Python
- Homepage:
- Size: 4.88 KB
- Stars: 3
- Watchers: 2
- Forks: 0
- Open Issues: 0