Add JAX array to registry

[timmens]

In this PR I add the Jax DeciveArray to the list of types. Tackles #17 .

[codecov]

Codecov Report

Merging #18 (dd95826) into main (708b392) will decrease coverage by 2.13%.
The diff coverage is 50.00%.

@@            Coverage Diff             @@
##             main      #18      +/-   ##
==========================================
- Coverage   95.30%   93.17%   -2.14%     
==========================================
  Files           9        9              
  Lines         405      425      +20     
==========================================
+ Hits          386      396      +10     
- Misses         19       29      +10     

Impacted Files	Coverage Δ
src/pybaum/registry_entries.py	82.05% <33.33%> (-6.36%)	⬇️
src/pybaum/equality.py	88.23% <60.00%> (-11.77%)	⬇️
src/pybaum/config.py	62.50% <66.66%> (+2.50%)	⬆️

---

Continue to review full report at Codecov.

Legend - Click here to learn more
Δ = absolute <relative> (impact), ø = not affected, ? = missing data
Powered by Codecov. Last update 708b392...dd95826. Read the comment docs.

[janosg]

I merge it in a new branch to be able to work on it because I don't have access to your fork.