Skip to main content

Eventpropjax

Eventax provides a JAX implementation of the EventProp algorithm using Diffrax and Equinox offering full autograd support, easy extension with custom neuron dynamics, and built-in delay training.

Features

  • Fully differentiable implementation via JAX and Diffrax
  • Easy extension with custom neuron model dynamics + learnable parameters
  • Support for (trainable) synnaptic delays.
  • GPU/TPU compatibility through JAX

📦 Installation

pip install eventax

Metadata

Release files for eventax 0.1.0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for eventax 0.1.0
File Size Uploaded
eventax-0.1.0.tar.gz 37.1 kB Details

Built distribution (wheel)

Table of built distributions (wheels) for eventax 0.1.0
File Interpreter ABI Platform
eventax-0.1.0-py3-none-any.whl Python 3 none any Details

Total release size: 79.1 kB

Release files / eventax-0.1.0.tar.gz

Download URL eventax-0.1.0.tar.gz
Size 37.1 kB
Tags Source
SHA-256 checksum
How to use checksums
26a2fff6a22a7c43ec0b4c0830dfadddffd8376f8d3027d161998ed65b7ea4e5
BLAKE2b-256 checksum
How to use checksums
a55506e488b089d478f26873be42c1587bf3cd5227d7ac1c258daaea83194ba9
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.2.0 CPython/3.12.9

Release files / eventax-0.1.0-py3-none-any.whl

Download URL eventax-0.1.0-py3-none-any.whl
Size 42.0 kB
Tags Python 3
SHA-256 checksum
How to use checksums
ffea7898584adab5a82eaeb5516dc0b314e1adaa24030064f0fe5f5ec3605686
BLAKE2b-256 checksum
How to use checksums
68d2f93bce0cbd2f6cde34e0485b73f9ae48db155d34762971fb553c26e75c7e
Upload date
Uploaded using Trusted Publishing?
What is trusted publishing?
No
Uploaded via twine/6.2.0 CPython/3.12.9

Release history Release notifications | RSS feed

This release

0.1.0 This release

2 release files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page