A Diffrax-based framework for continuous-time spiking neural networks
Project description
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
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
eventax-0.1.0.tar.gz
(37.1 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
eventax-0.1.0-py3-none-any.whl
(42.0 kB
view details)
File details
Details for the file eventax-0.1.0.tar.gz.
File metadata
- Download URL: eventax-0.1.0.tar.gz
- Upload date:
- Size: 37.1 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
26a2fff6a22a7c43ec0b4c0830dfadddffd8376f8d3027d161998ed65b7ea4e5
|
|
| MD5 |
4be6b81056ceed5b105ccb505ef17214
|
|
| BLAKE2b-256 |
a55506e488b089d478f26873be42c1587bf3cd5227d7ac1c258daaea83194ba9
|
File details
Details for the file eventax-0.1.0-py3-none-any.whl.
File metadata
- Download URL: eventax-0.1.0-py3-none-any.whl
- Upload date:
- Size: 42.0 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
ffea7898584adab5a82eaeb5516dc0b314e1adaa24030064f0fe5f5ec3605686
|
|
| MD5 |
869038177b304069eb5db43c3a65d3bc
|
|
| BLAKE2b-256 |
68d2f93bce0cbd2f6cde34e0485b73f9ae48db155d34762971fb553c26e75c7e
|
