Metadata-Version: 2.4
Name: decijax
Version: 0.0.1
Summary: Sequential decision-making in JAX.
Project-URL: Documentation, https://thomas-christie.github.io/decijax/
Project-URL: Issues, https://github.com/Thomas-Christie/decijax/issues
Project-URL: Source, https://github.com/Thomas-Christie/decijax
Author-email: Thomas Christie <thomashamish@hotmail.com>
License-Expression: MIT
License-File: LICENSE
Keywords: bayesian-optimisation,decision-making,gaussian-processes,jax
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Developers
Classifier: Intended Audience :: Science/Research
Classifier: Natural Language :: English
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.11
Classifier: Programming Language :: Python :: 3.12
Classifier: Programming Language :: Python :: 3.13
Classifier: Programming Language :: Python :: 3.14
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: Topic :: Scientific/Engineering :: Mathematics
Classifier: Typing :: Typed
Requires-Python: >=3.11
Requires-Dist: gpjax>=0.11.0
Requires-Dist: jax>=0.7.2
Requires-Dist: jaxlib>=0.7.2
Requires-Dist: jaxtyping>0.2.10
Requires-Dist: numpy>=2.0.0
Requires-Dist: numpyro>=0.19.0
Requires-Dist: scipy>=1.13.0
Description-Content-Type: text/markdown

# decijax

`decijax` is an early-stage library which aims to provide algorithms for a
variety of sequential decision-making problems. It currently provides implementations
of several acquisition functions for Bayesian optimisation, including
probability of improvement, expected improvement and Thompson sampling. The
implementations are built upon the JAX library, enabling automatic differentiation,
vectorisation, and just-in-time (JIT) compilation for high performance. This allows for efficient research, development, and deployment of decision-making agents.

> **⚠️ Warning**
>
> `decijax` is currently under active development, and the API is likely to
> change in the near future.

-----

## Table of Contents

- [decijax](#decijax)
  - [Table of Contents](#table-of-contents)
  - [Installation](#installation)
  - [Documentation](#documentation)
  - [Contributing](#contributing)
  - [License](#license)

## Installation

`decijax` requires Python 3.11 or later. Install the latest release from PyPI with:

```bash
pip install decijax
```

## Documentation

Available at [https://thomas-christie.github.io/decijax/](https://thomas-christie.github.io/decijax/).

## Contributing

Please refer to the [contributing guidelines](https://github.com/Thomas-Christie/decijax/blob/main/CONTRIBUTING.md) file for guidelines on how to contribute to the
project.

## License

`decijax` is distributed under the terms of the [MIT](https://spdx.org/licenses/MIT.html) license.
