PyPI page
Author:
PyTorch Team
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:
4,888
Downloads last week:
17,276
Downloads last month:
73,750
7-day smoothing