You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
18 lines
377 B
18 lines
377 B
[project]
|
|
name = "diffusion-points"
|
|
version = "0.1.0"
|
|
description = "Add your description here"
|
|
readme = "README.md"
|
|
requires-python = ">=3.11"
|
|
dependencies = [
|
|
"distrax>=0.1.5",
|
|
"einops>=0.8.1",
|
|
"flax>=0.10.6",
|
|
"jax[cuda12]>=0.6.0",
|
|
"numpy>=2.2.5",
|
|
"orbax>=0.1.9",
|
|
"seaborn>=0.13.2",
|
|
"tqdm>=4.67.1",
|
|
"tyro>=0.9.19",
|
|
"wandb>=0.19.10",
|
|
]
|
|
|