Port details on branch 2022Q2 |
- py-jax Differentiate, compile, and transform Numpy code
- 0.4.25 math =0 0.4.23Version of this port present on the latest quarterly branch.
- Maintainer: sunpoet@FreeBSD.org
- Port Added: 2020-07-09 18:10:24
- Last Update: 2024-03-09 14:06:10
- Commit Hash: 8d47cbf
- Also Listed In: python
- License: APACHE20
- WWW:
- https://github.com/google/jax
- Description:
- JAX is Autograd and XLA, brought together for high-performance machine learning
research.
With its updated version of Autograd, JAX can automatically differentiate native
Python and NumPy functions. It can differentiate through loops, branches,
recursion, and closures, and it can take derivatives of derivatives of
derivatives. It supports reverse-mode differentiation (a.k.a. backpropagation)
via grad as well as forward-mode differentiation, and the two can be composed
arbitrarily to any order.
What's new is that JAX uses XLA to compile and run your NumPy programs on GPUs
and TPUs. Compilation happens under the hood by default, with library calls
getting just-in-time compiled and executed. But JAX also lets you just-in-time
compile your own Python functions into XLA-optimized kernels using a
one-function API, jit. Compilation and automatic differentiation can be composed
arbitrarily, so you can express sophisticated algorithms and get maximal
performance without leaving Python. You can even program multiple GPUs or TPU
cores at once using pmap, and differentiate through the whole thing.
Dig a little deeper, and you'll see that JAX is really an extensible system for
composable function transformations. Both grad and jit are instances of such
transformations. Others are vmap for automatic vectorization and pmap for
single-program multiple-data (SPMD) parallel programming of multiple
accelerators, with more to come.
- ¦ ¦ ¦ ¦
- Manual pages:
- FreshPorts has no man page information for this port.
- pkg-plist: as obtained via:
make generate-plist - There is no configure plist information for this port.
- Dependency lines:
-
- ${PYTHON_PKGNAMEPREFIX}jax>0:math/py-jax@${PY_FLAVOR}
- To install the port:
- cd /usr/ports/math/py-jax/ && make install clean
- To add the package, run one of these commands:
- pkg install math/py-jax
- pkg install py39-jax
NOTE: If this package has multiple flavors (see below), then use one of them instead of the name specified above. NOTE: This is a Python port. Instead of py39-jax listed in the above command, you can pick from the names under the Packages section.- PKGNAME: py39-jax
- Package flavors (<flavor>: <package>)
- distinfo:
- TIMESTAMP = 1709390684
SHA256 (jax-0.4.25.tar.gz) = a8ee189c782de2b7b2ffb64a8916da380b882a617e2769aa429b71d79747b982
SIZE (jax-0.4.25.tar.gz) = 1543586
Packages (timestamps in pop-ups are UTC):
- Dependencies
- NOTE: FreshPorts displays only information on required and default dependencies. Optional dependencies are not covered.
- Build dependencies:
-
- py39-setuptools>=0 : devel/py-setuptools@py39
- py39-wheel>=0 : devel/py-wheel@py39
- python3.9 : lang/python39
- py39-build>=0 : devel/py-build@py39
- py39-installer>=0 : devel/py-installer@py39
- Test dependencies:
-
- python3.9 : lang/python39
- Runtime dependencies:
-
- py39-ml-dtypes>=0.2.0 : math/py-ml-dtypes@py39
- py39-numpy>=1.22,1 : math/py-numpy@py39
- py39-opt-einsum>=0 : math/py-opt-einsum@py39
- py39-scipy>=1.11.1 : science/py-scipy@py39
- python3.9 : lang/python39
- py39-importlib-metadata>=4.6 : devel/py-importlib-metadata@py39
- This port is required by:
- for Run
-
- math/py-flax
- science/meep
Configuration Options:
- No options to configure
- Options name:
- math_py-jax
- USES:
- python
- FreshPorts was unable to extract/find any pkg message
- Master Sites:
|