mirror of
https://github.com/cactus-compute/needle.git
synced 2026-10-02 05:04:32 +08:00
11 lines
129 B
Plaintext
11 lines
129 B
Plaintext
-r requirements.txt
|
|
|
|
# Training and checkpoint export dependencies
|
|
numpy
|
|
jax
|
|
jaxlib
|
|
flax>=0.10.2
|
|
optax
|
|
safetensors
|
|
sentencepiece
|