galacticdynamics / quaxed Goto Github PK View Code? Open in Web Editor NEW 3.0 3.0 1.0 169 KB Pre-`quaxify`d libraries for use in multiple dispatch over abstract array types in JAX. Home Page: https://quaxed.readthedocs.io/en/latest/ License: BSD 3-Clause "New" or "Revised" License Python 100.00% array array-api ducktyping jax multiple-dispatch Introduction ยท People ยท Discuss
jit where appropriate Figure out which functions can be jitted (with inline=True to let them be inlined in containing jit).
Rename repo (and PyPI) to `quaxed` Then we can have quaxed.array_api, quaxed.numpy, etc to cover the full namespace in one area.
More structured approach to adding the numpy namespace rather than the extremely inefficient piecemeal
Change repo name The real thing this repo adds is the quax wrapping. The name should better reflect this.
Add pytest-github-actions-annotate-failures to CI https://github.com/pytest-dev/pytest-github-actions-annotate-failures
Add tests unit tests test whether something can be more directly imported from jax.experimental.array_api. In particular, this repo adds jaxtyping + beartype runtime-checking support, which relies on type hints that are not in jax.experimental.array_api.
Vue.js ๐ Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
javascript JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Machine learning Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Facebook We are working to build community through open source technology. NB: members must have two-factor auth.