-
Notifications
You must be signed in to change notification settings - Fork 25
/
setup.py
38 lines (34 loc) · 1.15 KB
/
setup.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
"""
Copyright (c) Meta Platforms, Inc. and affiliates.
All rights reserved.
This source code is licensed under the BSD-style license found in the
LICENSE file in the root directory of this source tree.
"""
from setuptools import find_packages, setup
if __name__ == "__main__":
setup(
name="distributed_shampoo",
version="0.1.0",
description="PyTorch implementation of Distributed Shampoo",
license="BSD 3-clause",
packages=find_packages(),
py_modules=[
"matrix_functions",
"matrix_functions_types",
"optimizer_modules",
],
install_requires=[
"torch>=2.0.0",
"torchvision>=0.15.0",
],
python_requires=">=3.10",
classifiers=[
"Development Status :: 3 - Alpha",
"Intended Audience :: Science/Research",
"Intended Audience :: Developers",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
],
)