openretina

---

Welcome!

openretina is an open-source neural network toolkit for modeling retinal responses to visual stimuli. It provides pre-trained models for inference and interpretability, as well as components for training new retina models with your own data.

---

• Package repository

  ---

  Check out the main package repository on GitHub

• Read the paper

  ---

  openretina has a biorxiv pre-print!

• Install

  ---

  git clonepip

  Default, most flexible option. Comes with config files and notebooks.

  git clone git@github.com:open-retina/open-retina.git
  cd open-retina
  pip install -e .
  

  Simplest installation for using pre-trained models

  pip install openretina
  

• Read the documentation

  ---

  Browse detailed documentation on models, data loading, and in-silico experiments

---

Quick Start

import torch
from openretina.models import load_core_readout_from_remote

# Load a pre-trained model
model = load_core_readout_from_remote("hoefling_2024_base_low_res", "cpu")

# Run a forward pass with random input
responses = model.forward(torch.rand(model.stimulus_shape(time_steps=50)))
print(f"Model predicted responses shape: {responses.shape}")