felice-models/docs-site/docs/index.md

Felice

This project provides a JAX implementation of the different neuron models in felice

Overview

The framework is built on top of diffrax and leverages JAX's automatic differentiation for efficient simulation and training of analogue models.

Key Features

• Delay learning
• Non-linear neuron models
  • WereRabbit Neuron Model: Implementation of a dual-state oscillatory neuron model with bistable dynamics
  • FHN Neuron Model
  • Snowball Neuron Model

📦 Installation

Felice uses uv for dependency management. To install:

uv sync

CUDA Support (Optional)

For GPU acceleration with CUDA 13:

uv sync --extra cuda

See the examples directory for more detailed usage examples.