adding jax code

[b-biswas]

No description provided.

[codecov]

Codecov Report

❌ Patch coverage is 92.14660% with 60 lines in your changes missing coverage. Please review.
✅ Project coverage is 96.94%. Comparing base (1f51bd1) to head (1a99983).

Files with missing lines	Patch %	Lines
...d_metadetect/jaxify/tests/test_jax_deep_metacal.py	64.82%	51 Missing ⚠️
deep_field_metadetect/jaxify/jax_metacal.py	95.29%	8 Missing ⚠️
deep_field_metadetect/jaxify/observation.py	98.24%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main      #30      +/-   ##
==========================================
- Coverage   99.05%   96.94%   -2.11%     
==========================================
  Files          20       28       +8     
  Lines        1689     2421     +732     
==========================================
+ Hits         1673     2347     +674     
- Misses         16       74      +58     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.