pyproject.toml

[project]
name = "sbgm"
version = "0.0.21"
description = "Score-based Diffusion models in JAX."
readme = "README.md"
requires-python ="~=3.12"
license = {file = "LICENSE"}
authors = [
  {name = "Jed Homer", email = "jedhmr@gmail.com"},
]
keywords = [
    "jax", 
    "deep-learning", 
    "generative-models", 
    "diffusion-models",
    "score-based-diffusion"
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: MIT License",
    "Natural Language :: English",
    "Programming Language :: Python :: 3",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
urls = {repository = "https://github.com/homerjed/sbgm"}

dependencies = [
    'jax==0.4.33',
    'equinox==0.11.7',
    'diffrax==0.6.0',
    'optax==0.2.3',
    'beartype==0.19.0',
    'ml_collections',
    'numpy==1.26.4',
    'matplotlib==3.9.1',
    'einops>=0.8.0',
    'torch==2.4.0',
    'torchvision==0.19.0',
    'cloudpickle==3.0.0',
    'tqdm==4.66.4',
    'powerbox==0.8.1',
    'einx==0.3.0'
]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build]
include = ["sbgm/*"]