pyproject.toml

[build-system]
requires = ["flit_core >=3.2,<4"]
build-backend = "flit_core.buildapi"

[project]
name = "psgd-jax"
version = "0.2.9"
description = "An implementation of PSGD optimizer in JAX."
readme = { file = "README.md", content-type = "text/markdown" }
license = { file = "LICENSE" }
requires-python = ">=3.9"
authors = [
    { name = "Evan Walters" },
    { name = "Omead Pooladzandi" },
    { name = "Xi-Lin Li" },
]
keywords = [
    "python",
    "machine learning",
    "optimization",
    "jax",
]
classifiers = [
    "Environment :: Console",
    "Programming Language :: Python",
    "Intended Audience :: Developers",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
    "Intended Audience :: Science/Research",
    "Development Status :: 4 - Beta",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Software Development :: Libraries :: Python Modules",
]
dependencies = [
    "numpy",
    "matplotlib",
    "jax",
    "optax",
]

[project.urls]
homepage = "https://github.com/evanatyourservice/psgd_jax"
repository = "https://github.com/evanatyourservice/psgd_jax"