PyPI page
Author:
anton feldmann
Summary:
Comprehensive deep learning framework with state-of-the-art implementations: GPT, DeepSeek-V3 with MLA/MoE, YOLO, CenterNet, and specialized audio/vision models built on PyTorch Lightning
Latest version:
1.1.0
Required dependencies:
albumentations
|
any-llm-sdk
|
complexpytorch
|
huggingface-hub
|
lightning
|
loguru
|
numpy
|
psutil
|
py-cpuinfo
|
ray
|
toml
|
torch
|
torchinfo
|
torchmetrics
|
torchvision
|
torchviz
Optional dependencies:
albumentations
|
any-llm-sdk
|
autodoc-pydantic
|
black
|
furo
|
huggingface-hub
|
matplotlib
|
mlflow
|
motmetrics
|
mypy
|
myst-parser
|
opencv-python-headless
|
plotly
|
pre-commit
|
pygraphviz
|
pytest
|
ray
|
recommonmark
|
ruff
|
sphinx
|
sphinx-autodoc-typehints
|
tensorboard
|
torch
|
torchinfo
|
torchsummary
|
torchvision
|
torchviz
|
types-toml
|
wandb
|
zarr
Downloads last day:
3
Downloads last week:
14
Downloads last month:
178
7-day smoothing