JAX Random Projection Transformers
Using JAX to speed up sklearn's random projection transformers
Installation
Note: Installation with pip will install the CPU-only version of JAX
To use a GPU follow JAX's installation guide before installing jax-random_projections.
pip install jax-random_projections
Usage
from jax_random_projections.sparse import SparseRandomProjectionJAX
transfomer = SparseRandomProjectionJAX()
transfomer.fit_transform(X)
For the API documentation, refer to sklearn's SparseRandomProjection documentation.
The only difference is that jax-random_projections currently only supports xla.DeviceArray and doesn't support dense_output=False and y for fit()
This library currently only includes the SparseRandomProjection but a future release will also include GaussianRandomProjection.
jax-random_projections also includes SparseRandomProjectionJAXCached which uses a lru cache (maxsize=5) to speed up repeated calls by caching the random matrix for data with the same input dimension.
1 maintainer
Metadata
1 maintainer
Release files for jax-random-projections 1.0.1
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| jax-random_projections-1.0.1.tar.gz | 2.7 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| jax_random_projections-1.0.1-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 6.6 kB
Release files / jax-random_projections-1.0.1.tar.gz
| Download URL | jax-random_projections-1.0.1.tar.gz |
|---|---|
| Size | 2.7 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
7eef6e89bedfe44429c149bdc367934c201d4706cc364e640320976ecc7d2ab1
|
|
BLAKE2b-256 checksum How to use checksums |
0a59e9b52356e516d047a1a374e143519631a5adf3b99f055b04d90780eb6dc5
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.1.1 pkginfo/1.5.0.1 requests/2.23.0 setuptools/46.4.0 requests-toolbelt/0.9.1 tqdm/4.46.0 CPython/3.7.7
|
Release files / jax_random_projections-1.0.1-py3-none-any.whl
| Download URL | jax_random_projections-1.0.1-py3-none-any.whl |
|---|---|
| Size | 4.0 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
582fca45fcce4c80b8a2fd4879738a3a299d7ac672e08092d07fa307d06a8980
|
|
BLAKE2b-256 checksum How to use checksums |
7d4e126cd335f29e8e6e13468dcaadcd05bad4c0d9e1f7eebf2ff5fe764b9721
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/3.1.1 pkginfo/1.5.0.1 requests/2.23.0 setuptools/46.4.0 requests-toolbelt/0.9.1 tqdm/4.46.0 CPython/3.7.7
|
