ISTA-DASLab-Optimizers/pyproject.toml at main · IST-DASLab/ISTA-DASLab-Optimizers · GitHub

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
39
40
41
42
43
44
45
46
47
48
[build-system]
requires = ["setuptools", "wheel", "torch"]
build-backend = "setuptools.build_meta"

[project]
name='ista_daslab_optimizers'
version='1.1.14'
dependencies = [
    "torch", # >=2.3.1",
    "torchaudio", # >=2.3.1",
    "torchvision", #>=0.18.1",
    "numpy", # >=1.24.1",
    "wandb",#>=0.17.1",
    "gpustat",#>=1.1.1",
    "timm", # >=1.0.3",
    "einops", # >=0.7.0",
    "psutil", # >=5.9.8",
    "triton", # for Dion kernels
    "ista-daslab-optimizers-cuda",
#    "fast-hadamard-transform @ git+https://github.com/Dao-AILab/fast-hadamard-transform.git",
]
requires-python = '>= 3.8'
authors = [
    {name = "Ionut-Vlad Modoranu", email = "ionut-vlad.modoranu@ist.ac.at"}
]
maintainers = [
    {name = "Ionut-Vlad Modoranu", email = "ionut-vlad.modoranu@ist.ac.at"},
]
description = 'Deep Learning optimizers developed in the Distributed Algorithms and Systems group (DASLab) @ Institute of Science and Technology Austria (ISTA)'
readme = "README.md"
license = {file = "LICENSE"}
keywords = [
    "adaptive optimization",
    "deep learning",
    "low memory optimization",
]
classifiers = [
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.8",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "License :: OSI Approved :: MIT License",
]

[project.urls]
Repository = 'https://github.com/IST-DASLab/ISTA-DASLab-Optimizers'