search

ucl-bug / jaxdf

A JAX-based research framework for writing differentiable numerical simulators with arbitrary discretizations
GNU Lesser General Public License v3.0
117 stars 7 forks source link

removed jaxlib dependency #131

Closed astanziola closed 1 year ago

astanziola commented 1 year ago

Removing jaxlib from the dependencies.

Jax does not specifically list jaxlib in its dependencies due to its various hardware-specific versions (such as for CPU, GPU, TPU, and so on). Python's dependency management system lacks the necessary precision to handle this. This issue is resolved by JAX through its installation guidelines.

For more info: https://github.com/google/jax/discussions/16380