guilhem_lavaux/JaxPM
0
0
Fork 1
mirror of https://github.com/DifferentiableUniverseInitiative/JaxPM.git synced 2025-06-29 08:31:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Adds a trivial jaxpm implementation

Browse source
This commit is contained in:
EiffL 2022-02-13 21:36:03 +01:00
parent d4d0a03c79
commit afdfcfa644
5 changed files with 714 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

11
setup.py Normal file
View file

@ -0,0 +1,11 @@
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'],
)