Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
@bclarkson-code
bclarkson-code committed Jun 9, 2024
1 parent 151aa64 commit 522e505
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ The file `train_smol_gpt.py` trains a 49M parameter, GPT-2 style language model

The entire library, from the automatic differentiation engine to a GPT, is written in ~4500 lines of python + numpy code.

Using [CuPY](https://cupy.dev/), all Tricycle code can run on a either GPU or a CPU.
Using [CuPY](https://cupy.dev/), all Tricycle code can run on either a Cuda-capable GPU or a CPU.


## Table of Contents
Expand Down

0 comments on commit 522e505

Please sign in to comment.