FlyCore for Python and PyTorch

The reference engine. Sparse connectome forward pass in PyTorch, training loops included, CUDA and Apple silicon.

pip install flycore and load any weight set from Hugging Face. The same package trains new weights: bring an environment or a dataset, keep the brain.

pip
pip install flycore

Runtime: PyTorch 2.x, torch.sparse with custom sparse backward; CUDA and Apple silicon

Available weights
Pricing

Every weight set is free up to 100k monthly active devices per platform. Unlimited inference per user.