JAX_ENUMS

Installation | Examples | Cite

A Jax-compatible enumerable.

Installation

pip install jax_enums

Example

class Foo(Enumerable):
    BAR = 0
    BAZ = 1

def f(array: jax.Array, enumerable: Enum) -> jax.Array:
    return array[enumerable.value]

array = jnp.zeros((2, 2))
enumerable = Foo.BAR

f(array, enumerable)
jax.jit(f)(array, enumerable)

Cite

@misc{pignatelli2023jax_enums,
  author = {Pignatelli, Eduardo},
  title = {JAX_ENUMS: JAX-compatible enumerations},
  year = {2023},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/epignatelli/jax_enums}}
  }

Metadata

Release files for jax-enums 0.1.2

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for jax-enums 0.1.2
File	Size	Uploaded	Provenance
jax_enums-0.1.2.tar.gz	15.9 kB	Jul 27, 2023	Without provenance	Details

Built distribution (wheel)

Table of built distributions (wheels) for jax-enums 0.1.2
File	Interpreter	ABI	Platform	Reset
jax_enums-0.1.2-py2.py3-none-any.whl 12.9 kB Jul 27, 2023 Without provenance	Python 2, Python 3	none	any	Details

Total release size: 28.8 kB

Release files / jax_enums-0.1.2.tar.gz

Download URL	jax_enums-0.1.2.tar.gz
Size	15.9 kB
Tags	Source
SHA-256 checksum How to use checksums	`75391a59e1f5be3a976715306182d22331d74c001ae3b580e95a88c03d58dabf`
BLAKE2b-256 checksum How to use checksums	`c97e86dadf746bba5e0c7cfd08db797dfd9e52c9bbe904ee74998727cac17381`
Upload date	Jul 27, 2023
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/4.0.2 CPython/3.11.4`

Release files / jax_enums-0.1.2-py2.py3-none-any.whl

Download URL	jax_enums-0.1.2-py2.py3-none-any.whl
Size	12.9 kB
Tags	Python 2 Python 3
SHA-256 checksum How to use checksums	`e9c9374789f9adb5874013c80e2be411af0ef1442cbc3c3c0bb0ac89de7f2f2c`
BLAKE2b-256 checksum How to use checksums	`c9744cc28410c9f025ccb575123f35941f2812e67afd975f42b99682152e939e`
Upload date	Jul 27, 2023
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/4.0.2 CPython/3.11.4`

Security for jax-enums 0.1.2

Report project as malware

Provenance

Provenance describes where a file came from. On PyPI, provenance is shared via attestations, which provide a verifiable record of the build or publishing details. View details, limitations and caveats.

No provenance found

None of the files in this release were uploaded with provenance metadata.

A note on package security

PyPI is a repository for distributing packages, not a service for auditing them. While third-party security researchers and our staff do their best to detect and remove malicious packages, PyPI cannot guarantee the quality, security or suitability of any given package.

To learn more about protecting yourself from malicious actors, visit our package security documentation.