2024-07-09 14:54:34 -04:00
|
|
|
from setuptools import find_packages, setup
|
2022-02-13 21:36:03 +01:00
|
|
|
|
|
|
|
|
setup(
|
|
|
|
|
name='JaxPM',
|
|
|
|
|
version='0.0.1',
|
|
|
|
|
url='https://github.com/DifferentiableUniverseInitiative/JaxPM',
|
|
|
|
|
author='JaxPM developers',
|
|
|
|
|
description='A dead simple FastPM implementation in JAX',
|
2024-07-09 14:54:34 -04:00
|
|
|
packages=find_packages(),
|
2022-02-13 21:36:03 +01:00
|
|
|
install_requires=['jax', 'jax_cosmo'],
|
2024-07-09 14:54:34 -04:00
|
|
|
)
|
