Metadata-Version: 2.1
Name: rationalnets
Version: 0.0.1
Summary: JAX/Flax implementation of rational neural nets
Home-page: https://github.com/yonesuke/RationalNets
Author: yonesuke
Author-email: 13e.e.c.13@gmail.com
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.7
Description-Content-Type: text/markdown
License-File: LICENSE
Requires-Dist: jax
Requires-Dist: flax

# RationalNets
JAX/Flax implementation of rational neural nets.

Original
- paper: Nicolas Boullé, Yuji Nakatsukasa, and Alex Townsend, [Rational neural networks](https://arxiv.org/abs/2004.01902), arXiv preprint arXiv:2004.01902 (2020).
- github: https://github.com/NBoulle/RationalNets


## Installation
RationalNets can be installed with pip directly from GitHub, with the following command:
```
pip install git+https://github.com/yonesuke/RationalNets.git
```

## QuickStart
```python
import jax.numpy as jnp
from jax import random
from rationalnets import RationalMLP

model = RationalMLP([12, 8, 4])
batch = jnp.ones((32, 10))
variables = model.init(random.PRNGKey(0), batch)
output = model.apply(variables, batch)
```
