Metadata-Version: 2.4
Name: jux-numpy
Version: 0.1.0
Summary: AI Mathematics + Tensor Operations Toolkit built on JAX
Author: Admin
Classifier: Development Status :: 3 - Alpha
Classifier: Intended Audience :: Science/Research
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.10
Classifier: License :: OSI Approved :: MIT License
Requires-Python: >=3.8
Description-Content-Type: text/markdown
Requires-Dist: jax
Requires-Dist: jaxlib
Dynamic: author
Dynamic: requires-python


# Jux-Numpy 🚀

Jux-Numpy is an AI mathematics and tensor operations toolkit built on top of JAX.

## Features

- AI Math operations (entropy, KL divergence, mutual information)
- ML metrics (MAE, MSE, RMSE)
- Probability tools (softmax, entropy)
- Vector similarity tools (cosine, euclidean)
- JAX-powered performance backend

## Example

```python
import jux_numpy as jux

x = jux.array([1,2,3])
y = jux.softmax(x)

print(y)
