JaxPM/setup.py
2024-07-09 14:54:34 -04:00

11 lines
328 B
Python

from setuptools import find_packages, setup
setup(
name='JaxPM',
version='0.0.1',
url='https://github.com/DifferentiableUniverseInitiative/JaxPM',
author='JaxPM developers',
description='A dead simple FastPM implementation in JAX',
packages=find_packages(),
install_requires=['jax', 'jax_cosmo'],
)