/jina-gsoc-prework

Contains code for the prework done on Jax support in docarray v2 project for GSOC 2023

Primary LanguagePython

Jina DocArray v2 - Jax integration #21

Draft Proposal: Google docs

  • Please Note: This is not official and at the same time there could be a bugs in the code.
  • Also, this is intended to be only for the issue: #21
  • Need to work on the AbstractJaxComputationalBackend which will located at docarray/computation/ in feat-rewrite-v2 branch
  • Sole purpose of this is to get familiarized with the issue and learn the Jax methods and attributes before GSOC coding period begins