Implement a classifier on Jax framework

[minhitbk]

Description

This PR is to implement a classifier on the Jax framework.

Type of change

Please check all relevant options.

• Improvement (non-breaking)
• Bug fix (non-breaking)
• New feature (non-breaking)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• This change requires a documentation update

Testing

Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.

• Test A
• Test B

Test Configuration:

• OS
• Python version
• ART version or commit number
• TensorFlow / Keras / PyTorch / MXNet version

Checklist

• My code follows the style guidelines of this project
• I have performed a self-review of my own code
• I have commented my code
• I have made corresponding changes to the documentation
• My changes generate no new warnings
• I have added tests that prove my fix is effective or that my feature works
• New and existing unit tests pass locally with my changes

[codecov-commenter]

Codecov Report

Merging #1360 (448958a) into dev_1.9.0 (4f41ea7) will decrease coverage by 0.40%.
The diff coverage is 0.00%.

@@              Coverage Diff              @@
##           dev_1.9.0    #1360      +/-   ##
=============================================
- Coverage      90.31%   89.90%   -0.41%     
=============================================
  Files            240      244       +4     
  Lines          19672    19759      +87     
  Branches        3490     3495       +5     
=============================================
- Hits           17766    17764       -2     
- Misses          1109     1198      +89     
  Partials         797      797              

Impacted Files	Coverage Δ
art/experimental/__init__.py	0.00% <0.00%> (ø)
...experimental/estimators/classification/__init__.py	0.00% <0.00%> (ø)
art/experimental/estimators/classification/jax.py	0.00% <0.00%> (ø)
art/experimental/estimators/jax.py	0.00% <0.00%> (ø)
art/utils.py	91.13% <ø> (-0.02%)	⬇️
art/attacks/evasion/boundary.py	92.72% <0.00%> (-1.22%)	⬇️
art/defences/detector/poison/activation_defence.py	85.18% <0.00%> (+0.33%)	⬆️

[lgtm-com]

This pull request introduces 1 alert when merging 064e369 into 2717467 - view on LGTM.com

new alerts:

• 1 for Unused import