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.4.0
Required dependencies:
attn-gym
|
datasets
|
einops
|
grain
|
pillow
|
pyarrow
|
renderers
|
safetensors
|
spmd-types
|
tensorboard
|
tokenizers
|
torch-checkpointing
|
torch-remat
|
wandb
Optional dependencies:
av
|
einops
|
expecttest
|
numpy
|
opentelemetry-exporter-otlp-proto-http
|
opentelemetry-sdk
|
pillow
|
portpicker
|
pre-commit
|
pygtrie
|
pyrefly
|
pytest
|
pytest-cov
|
sentencepiece
|
torchmonarch
|
torchvision
|
transformers
Downloads last day:
2,829
Downloads last week:
51,892
Downloads last month:
73,750
7-day smoothing