JAX NN library.
Project description
The JAX NN Library.
🛠️ Installation
pip install serket
📖 Description
serket
aims to be the most intuitive and easy-to-use Neural network library in JAX.serket
is built on top ofpytreeclass
serket
currently implementsLinear
,FNN
Dropout
Sequential
Lambda
⏩ Quick Example
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
serket-0.0.0rc0.tar.gz
(5.4 kB
view hashes)
Built Distribution
Close
Hashes for serket-0.0.0rc0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | ab8859296a07ed25b4281071d8d1fb0eeb3860f67242c7f1a7abe3d55517f1c1 |
|
MD5 | e789c7104cc749db0bacd4579cc88fb1 |
|
BLAKE2b-256 | 896f629ff8e2d32658d5ff7402d96603a2b2dfbf7bcf9bc2ca3905326e8c92f9 |