Helmholtz-AI-Energy / DLRT

DLRT

TODO:

• transformers
• resnet50 imagenet tests
• autocast
• test jit
• graphs
  • probably need to set up the rank adaption to avoid changing the memory until the graph is rebuilt
• set up checkpointing/loading (trainer.py)

Collection of DLRT methods in PyTorch for easy usage

• GitHub repo: https://github.com/coquelin77/DLRT.git
• Documentation: https://DLRT.readthedocs.io
• Free software: BSD

Features

• TODO

Quickstart

TODO

Credits

This package was created with Cookiecutter and the fedejaure/cookiecutter-modern-pypackage project template.