Skip to content

NN Template

CI CI Release Code style: black

"We demand rigidly defined areas of doubt and uncertainty."


cookiecutter https://github.com/grok-ai/nn-template

asciicast


Generic cookiecutter template to bootstrap PyTorch projects and to avoid writing boilerplate code to integrate:

  • PyTorch Lightning, lightweight PyTorch wrapper for high-performance AI research.
  • Hydra, a framework for elegantly configuring complex applications.
  • Hugging Face Datasets,a library for easily accessing and sharing datasets.
  • Weights and Biases, organize and analyze machine learning experiments. (educational account available)
  • Streamlit, turns data scripts into shareable web apps in minutes.
  • MkDocs and Material for MkDocs, a fast, simple and downright gorgeous static site generator.
  • DVC, track large files, directories, or ML models. Think "Git for data".
  • GitHub Actions, to run the tests, publish the documentation and to PyPI automatically.
  • Python best practices for developing and publishing research projects.

cookiecutter

This is a parametrized template that uses cookiecutter. Install cookiecutter with:

pip install cookiecutter