PyPI page
Home page
Author:
None
License:
BSD 3-Clause License
(c) Meta Platforms, Inc. and affiliates.
Redistribution and use in source and binary forms, with or without modification,
are permitted p...
Summary:
A PyTorch native platform for training generative AI models
Latest version:
0.3.0
Required dependencies:
datasets
|
einops
|
pillow
|
safetensors
|
spmd_types
|
tensorboard
|
tokenizers
|
torch_checkpointing
|
torchdata
|
tyro
|
wandb
Optional dependencies:
expecttest
|
pre-commit
|
pyrefly
|
pytest
|
pytest-cov
|
torch_remat
Downloads last day:
2,996
Downloads last week:
16,458
Downloads last month:
116,036