PyPI page
Home page
Author:
Sebastian Gabarain
License:
Apache-2.0
Summary:
Automatically optimize and prepare language models for TPU distributed training.
Latest version:
0.0.1.2
Required dependencies:
accelerate
|
bitsandbytes
|
datasets
|
jsonlines
|
langchain
|
peft
|
sentencepiece
|
tensorflow-cpu
|
torch
|
torch-xla
|
transformers
|
wandb
|
zstandard
Downloads last day:
8
Downloads last week:
25
Downloads last month:
35