Add initial code

This commit is contained in:
Igor Babuschkin
2024-03-14 15:03:58 -07:00
parent 5aabc78af1
commit be76c959fa
11 changed files with 2552 additions and 2 deletions

5
requirements.txt Normal file
View File

@ -0,0 +1,5 @@
dm_haiku==0.0.12
-f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
jax[cuda12_pip]==0.4.25
numpy==1.26.4
sentencepiece==0.2.0