Project description

Refx

A reference system for jax

What is Refx?

Refx is a library that allows you to define pydags (directed acyclic graphs) on top of JAX's pytrees via a simple reference system. It enable two key features:

Shared parameters / modules
Tractable mutability

Why Refx?

Functional systems like flax and haiku are powerful but add a lot of complexity that is often transfered to the user. On the other hand, pytree-based systems like equinox are simpler but lack the ability to share parameters and modules.

Refx aims to create a system that can be used to build neural networks libraries that has the simplicity of pytree-based systems while also having the power of functional systems.

Installation

pip install refx

Usage

Project details

These details have not been verified by PyPI

View statistics for this project via Libraries.io, or by using our public dataset on Google BigQuery

Release history Release notifications | RSS feed

0.0.3

Apr 25, 2023

0.0.2

Apr 10, 2023

This version

0.0.1

Apr 10, 2023

0.0.0

Mar 15, 2023

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

refx-0.0.1.tar.gz (7.6 kB view hashes)

Uploaded Apr 10, 2023 Source

Built Distribution

refx-0.0.1-py3-none-any.whl (8.6 kB view hashes)

Uploaded Apr 10, 2023 Python 3

Hashes for refx-0.0.1.tar.gz

Hashes for refx-0.0.1.tar.gz
Algorithm	Hash digest
SHA256	`9051237ab6a5d4c81d9c3d5787b5d722c13da9c39ac4a0f49c895f7bb328cd04`
MD5	`2c4fbfa89c55f4c6bcf169494fde7d94`
BLAKE2b-256	`3a33a517a253bb851c38f2c15fe1144f441bcc86afc496e9307ac55dabcee83d`

Hashes for refx-0.0.1-py3-none-any.whl

Hashes for refx-0.0.1-py3-none-any.whl
Algorithm	Hash digest
SHA256	`a0977185ceedf7fce712ccb6f870d4530a9dd0da2a2093f4cef2a76f6b6616f7`
MD5	`ca0072436538fdb6f140b9300ec33af0`
BLAKE2b-256	`8470677374682f364ae57a13ba1aeb65fc253c8ede58f8ffe6fbff269c7ab753`