Metadata-Version: 2.1
Name: jax-toolkit
Version: 0.1.0
Summary: A collection of jax functions to help with common machine/deep learning related functionality.
Home-page: https://github.com/asmith26/jax_toolkit.git
Author: asmith26
License: Apache-2.0
Description: # jax_toolkit
        
        A collection of jax functions to help with common machine/deep learning related functionality. 
        
        [Documentation](https://asmith26.github.io/jax_toolkit/), [PyPi](https://pypi.org/project/jax-toolkit/)
        
        This library currently contains the basics for a number of losses and metrics. We intend to add more complexity and functionality as and when it's needed - of course contributions/pull requests/bug reports etc. are very welcome if you discover problems or need something that is currently missing.
        
Platform: UNKNOWN
Classifier: Development Status :: 2 - Pre-Alpha
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3
Requires-Python: >=3.7
Description-Content-Type: text/markdown
Provides-Extra: losses_utils
