Metadata-Version: 2.1
Name: jaxopt
Version: 0.1
Summary: Hardware accelerated, batchable and differentiable optimizers in JAX.
Home-page: https://github.com/google/jaxopt
Author: Google LLC
Author-email: no-reply@google.com
License: Apache 2.0
Description: # JAXopt
        
        [**Installation**](#installation)
        | [**Documentation**](https://jaxopt.github.io)
        | [**Examples**](https://github.com/google/jaxopt/tree/main/examples)
        | [**Cite us**](#citeus)
        
        Hardware accelerated, batchable and differentiable optimizers in JAX.
        
        - **Hardware accelerated:** our implementations run on GPU and TPU, in addition
          to CPU.
        - **Batchable:** multiple instances of the same optimization problem can be
          automatically vectorized using JAX's vmap.
        - **Differentiable:** optimization problem solutions can be differentiated with
          respect to their inputs either implicitly or via autodiff of unrolled
          algorithm iterations.
        
        ## Installation<a id="installation"></a>
        
        To install the latest release of JAXopt, use the following command:
        
        ```bash
        $ pip install jaxopt
        ```
        
        To install the **development** version, use the following command instead:
        
        ```bash
        $ pip install git+https://github.com/google/jaxopt
        ```
        
        Alternatively, it can be be installed from sources with the following command:
        
        ```bash
        $ python setup.py install
        ```
        
        ## Cite us<a id="citeus"></a>
        
        Our implicit differentiation framework is described in this
        [paper](https://arxiv.org/abs/2105.15183). To cite it:
        
        ```
        @article{jaxopt_implicit_diff,
          title={Efficient and Modular Implicit Differentiation},
          author={Blondel, Mathieu and Berthet, Quentin and Cuturi, Marco and Frostig, Roy 
            and Hoyer, Stephan and Llinares-L{\'o}pez, Felipe and Pedregosa, Fabian 
            and Vert, Jean-Philippe},
          journal={arXiv preprint arXiv:2105.15183},
          year={2021}
        }
        ```
        
        ## Disclaimer
        
        JAXopt is an open source project maintained by a dedicated team in Google Research, but is not an official Google product.
Keywords: optimization,root finding,implicit differentiation,jax
Platform: UNKNOWN
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: License :: OSI Approved :: Apache Software License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.7
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Description-Content-Type: text/markdown
