File size: 175 Bytes
19712a8
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
jax[cuda12]
flax==0.9.0
flash_attn_jax
torch
torchvision
opencv-python-headless
einops
huggingface_hub
transformers
tokenizers
sentencepiece
fire
invisible-watermark
ml-dtypes