flashbax

View on PyPIReverse Dependencies (3)

0.1.2 flashbax-0.1.2-py3-none-any.whl

Wheel Details

Project: flashbax
Version: 0.1.2
Filename: flashbax-0.1.2-py3-none-any.whl
Download: [link]
Size: 73014
MD5: afb3022cd8fecf970859b98fcee98311
SHA256: ac50b2580808ce63787da0ae240db14986e3404ade98a33335e6d7a5efe84859
Uploaded: 2024-03-12 14:00:21 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: flashbax
Version: 0.1.2
Summary: Flashbax is an experience replay library oriented around JAX. Tailored to integrate seamlessly with JAX's Just-In-Time (JIT) compilation.
Author-Email: InstaDeep <hello[at]instadeep.com>
Project-Url: Homepage, https://github.com/instadeepai/flashbax
Project-Url: Bug Tracker, https://github.com/instadeepai/flashbax/issues
Project-Url: Repository, https://github.com/instadeepai/flashbax.git
Keywords: jax,memory,python,reinforcement-learning
Classifier: Development Status :: 2 - Pre-Alpha
Classifier: Environment :: Console
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Operating System :: OS Independent
Classifier: Programming Language :: Python :: 3.9
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Software Development :: Libraries :: Python Modules
Requires-Python: >=3.9
Requires-Dist: chex (>=0.1.8)
Requires-Dist: flax (>=0.6.11)
Requires-Dist: jax (>=0.4.10)
Requires-Dist: jaxlib (>=0.4.10)
Requires-Dist: numpy (>=1.19.5)
Requires-Dist: tensorstore (>=0.1.51)
Requires-Dist: typing-extensions (>=4.6.0)
Requires-Dist: black (>=23.7.0); extra == "dev"
Requires-Dist: coverage (>=6.5.0); extra == "dev"
Requires-Dist: flake8 (>=6.0.0); extra == "dev"
Requires-Dist: isort (>=5.12.0); extra == "dev"
Requires-Dist: mkdocs-git-revision-date-plugin (==0.3.2); extra == "dev"
Requires-Dist: mkdocs-include-markdown-plugin (==6.0.1); extra == "dev"
Requires-Dist: mkdocs-material (==9.4.2); extra == "dev"
Requires-Dist: mkdocs-mermaid2-plugin (==1.1.1); extra == "dev"
Requires-Dist: mkdocs (==1.5.3); extra == "dev"
Requires-Dist: mkdocstrings[python] (==0.23.0); extra == "dev"
Requires-Dist: mknotebooks (==0.8.0); extra == "dev"
Requires-Dist: mypy (>=0.982); extra == "dev"
Requires-Dist: pre-commit (>=2.20.0); extra == "dev"
Requires-Dist: pytest-cov (>=4.00); extra == "dev"
Requires-Dist: pytest-xdist (>=3.0.2); extra == "dev"
Requires-Dist: pytest (>=7.4.2); extra == "dev"
Requires-Dist: distrax; extra == "examples"
Requires-Dist: dm-haiku; extra == "examples"
Requires-Dist: gymnasium; extra == "examples"
Requires-Dist: jumanji; extra == "examples"
Requires-Dist: jupyterlab; extra == "examples"
Requires-Dist: matplotlib; extra == "examples"
Requires-Dist: optax; extra == "examples"
Requires-Dist: rlax; extra == "examples"
Requires-Dist: seaborn; extra == "examples"
Provides-Extra: dev
Provides-Extra: examples
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 21357 characters]

WHEEL

Wheel-Version: 1.0
Generator: hatchling 1.21.1
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
flashbax/__init__.py sha256=7B8t2lTX5FpJeAFG7TpPF2iJu8jUOaUDMnp31I_NXNM 950
flashbax/conftest.py sha256=cqO-pmPWsyf2vqcKxlPNvvWPZ18aKMZzyFPlo_VihoU 804
flashbax/utils.py sha256=4xqVkoFOpFpu3-y6u5SMGn0MB8VqJbSGzTqbDUNPGyk 2637
flashbax/utils_test.py sha256=zQyK1uE-aKF166OBPpGgdOOw09fzCfVVTlni91SxKH0 1528
flashbax/buffers/__init__.py sha256=0PglBvdJBEHW8mkNaUnekf9IqfzvDT_j0XZZBilrf8U 1038
flashbax/buffers/conftest.py sha256=Un0My7fiRZR4d5aUquasHcQ2yL1sDaTp6RhrKHmEcUw 1452
flashbax/buffers/flat_buffer.py sha256=SDQGPtclDnYo2Ee6gvs_G71rE3slh_TVAx7-mv_T3uc 6688
flashbax/buffers/flat_buffer_test.py sha256=pKkZDtMWZQWov3GU-HxfcZpexbVKa8GwGFK-hWvMPQE 8235
flashbax/buffers/item_buffer.py sha256=qb7MyfGeMs0toOu-T0V0PPLEYx-9QRQ7NyYLE2gqw-k 5221
flashbax/buffers/item_buffer_test.py sha256=oYT-NrJrqTP8mrGKhZxkJMjO4Wxdh_lY8qHkxsS-yOA 8129
flashbax/buffers/prioritised_flat_buffer.py sha256=tp2BtivJFn7qCg9l51xh57AO8LwwQpG0-v3Kx9L_kwg 5402
flashbax/buffers/prioritised_flat_buffer_test.py sha256=3igp6eUk1EriNAuvmw3G8XB4iEHW4YBtTIAFt2d2zas 11460
flashbax/buffers/prioritised_trajectory_buffer.py sha256=gvOtAu-ho_4JrlAhYjK-LL-Mewmf5TTvoOdWJ5v5jnc 35363
flashbax/buffers/prioritised_trajectory_buffer_test.py sha256=gNCFzcHx1RzgpbKf5qhluHfGvVcj2DNfR3HOrnN5wik 21094
flashbax/buffers/sum_tree.py sha256=5cEhvSjWpRxj9LtD2Y6YBzuxSWawLOYccfPIxOZNVIw 15093
flashbax/buffers/sum_tree_test.py sha256=k84sLcl04DMwH_8cYEHYnNjp0NZ2iW3y0xboTuUazBc 10807
flashbax/buffers/trajectory_buffer.py sha256=X7alKdVltSlB_77id2V8qcPG2SzNlliShbgdU7Iu16g 26439
flashbax/buffers/trajectory_buffer_test.py sha256=bFkY00J8c4fSFfkCPe_Y1w-wRIEV5xxFg5rO4CvU1es 13380
flashbax/buffers/trajectory_queue.py sha256=d52rJ4TCXHr1HPVhLObRermWeUlLMhJfOcfK09qcauo 15581
flashbax/buffers/trajectory_queue_test.py sha256=vIWWqJQH9EZ1bx_kyBbE6tslcE9ac_J1Z_kBd7JDWXA 15490
flashbax/vault/__init__.py sha256=ToKFBTV0fbjESxB1QU5xOV-DFubzOdnuzC8Q2x1RcvE 638
flashbax/vault/vault.py sha256=cgkbS7rlBiKbYyNJrqFJ2k5-xHa3PCeouMO_ZOYJM7w 18988
flashbax/vault/vault_test.py sha256=bqmiiGm9W0U8jZBddYn0Sn0qI_pSI_Lx-din5PqNzZI 6213
flashbax-0.1.2.dist-info/METADATA sha256=uFha9JTS1qz9fnhlicst-lidMSWzXxNfhW2yvmpC-5c 24013
flashbax-0.1.2.dist-info/WHEEL sha256=TJPnKdtrSue7xZ_AVGkp9YXcvDrobsjBds1du3Nx6dc 87
flashbax-0.1.2.dist-info/licenses/LICENSE sha256=iqEbL59b0yawpfQCfEqAV3CmvMRlSjtjKI6teaSoLMg 11343
flashbax-0.1.2.dist-info/RECORD