JaxPM/setup.py

11 lines
331 B
Python
Raw Permalink Normal View History

2022-02-13 21:36:03 +01:00
from setuptools import setup, find_packages
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'],
)