Pedro Cuenca
commited on
Commit
•
577b22f
1
Parent(s):
478aae4
Fix typo.
Browse files
README.md
CHANGED
|
@@ -20,7 +20,7 @@ This model's architecture is a simplification of the original, and leverages pre
|
|
| 20 |
|
| 21 |
### Components of the Architecture
|
| 22 |
|
| 23 |
-
The system relies
|
| 24 |
|
| 25 |
The main components of the architecture include:
|
| 26 |
|
|
|
|
| 20 |
|
| 21 |
### Components of the Architecture
|
| 22 |
|
| 23 |
+
The system relies on the Flax/JAX infrastructure, which are ideal for TPU training. TPUs are not required, both Flax and JAX run very efficiently on GPU backends.
|
| 24 |
|
| 25 |
The main components of the architecture include:
|
| 26 |
|