NVIDIA / TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs, to provide better performance with lower memory utilization in both training and inference.
https://docs.nvidia.com/deeplearning/transformer-engine/user-guide/index.html
Apache License 2.0
1.85k stars 309 forks source link

Use pyproject.toml to specify build requirements #1061

Open ksivaman opened 2 months ago

ksivaman commented 2 months ago

Description

setup_requires is deprecated (pep ref) and will be removed in setuptools v75.0.0. This PR uses the newer pyproject.toml to specify build time dependencies. We can consider gradually moving over other functionality to pyproject.toml as required.

Type of change

Changes

Please list the changes introduced in this PR:

Checklist:

ksivaman commented 2 months ago

/te-ci