{"id":25594416,"url":"https://github.com/chaobrain/braintaichi","last_synced_at":"2026-02-12T12:56:16.164Z","repository":{"id":254681875,"uuid":"847084578","full_name":"chaobrain/braintaichi","owner":"chaobrain","description":"Leveraging Taichi Lang to customize brain dynamics operators.","archived":false,"fork":false,"pushed_at":"2025-02-06T08:49:08.000Z","size":1839,"stargazers_count":0,"open_issues_count":8,"forks_count":0,"subscribers_count":2,"default_branch":"main","last_synced_at":"2025-02-06T09:36:15.549Z","etag":null,"topics":["brain-dynamics","event-driven-computation","sparse-computation"],"latest_commit_sha":null,"homepage":"https://braintaichi.readthedocs.io","language":"Python","has_issues":true,"has_wiki":null,"has_pages":null,"mirror_url":null,"source_name":null,"license":"apache-2.0","status":null,"scm":"git","pull_requests_enabled":true,"icon_url":"https://github.com/chaobrain.png","metadata":{"files":{"readme":"README.md","changelog":"changelog.md","contributing":"CONTRIBUTING.md","funding":null,"license":"LICENSE","code_of_conduct":"CODE_OF_CONDUCT.md","threat_model":null,"audit":null,"citation":null,"codeowners":null,"security":"SECURITY.md","support":null,"governance":null,"roadmap":null,"authors":null,"dei":null,"publiccode":null,"codemeta":null}},"created_at":"2024-08-24T19:58:48.000Z","updated_at":"2025-02-06T08:49:09.000Z","dependencies_parsed_at":"2024-09-12T20:26:46.259Z","dependency_job_id":"928617dc-3acf-4ac9-9488-e939441c54ed","html_url":"https://github.com/chaobrain/braintaichi","commit_stats":null,"previous_names":["chaoming0625/braintaichi","chaobrain/braintaichi"],"tags_count":0,"template":false,"template_full_name":null,"repository_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/chaobrain%2Fbraintaichi","tags_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/chaobrain%2Fbraintaichi/tags","releases_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/chaobrain%2Fbraintaichi/releases","manifests_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories/chaobrain%2Fbraintaichi/manifests","owner_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners/chaobrain","download_url":"https://codeload.github.com/chaobrain/braintaichi/tar.gz/refs/heads/main","host":{"name":"GitHub","url":"https://github.com","kind":"github","repositories_count":240002137,"owners_count":19732164,"icon_url":"https://github.com/github.png","version":null,"created_at":"2022-05-30T11:31:42.601Z","updated_at":"2022-07-04T15:15:14.044Z","host_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub","repositories_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repositories","repository_names_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/repository_names","owners_url":"https://repos.ecosyste.ms/api/v1/hosts/GitHub/owners"}},"keywords":["brain-dynamics","event-driven-computation","sparse-computation"],"created_at":"2025-02-21T11:00:34.313Z","updated_at":"2026-02-12T12:56:16.153Z","avatar_url":"https://github.com/chaobrain.png","language":"Python","readme":"# Leveraging Taichi Lang to Customize Brain Dynamics Operators\n\n\n\u003cp align=\"center\"\u003e\n  \t\u003cimg alt=\"Header image of braintaichi.\" src=\"https://github.com/chaoming0625/braintaichi/blob/main/docs/_static/braintaichi.png\" width=50%\u003e\n\u003c/p\u003e \n\n\n\n\u003cp align=\"center\"\u003e\n\t\u003ca href=\"https://pypi.org/project/braintaichi/\"\u003e\u003cimg alt=\"Supported Python Version\" src=\"https://img.shields.io/pypi/pyversions/braintaichi\"\u003e\u003c/a\u003e\n\t\u003ca href=\"https://github.com/chaoming0625/braintaichi/blob/main/LICENSE\"\u003e\u003cimg alt=\"LICENSE\" src=\"https://img.shields.io/badge/License-Apache%202.0-blue.svg\"\u003e\u003c/a\u003e\n    \u003ca href='https://braintaichi.readthedocs.io/en/latest/?badge=latest'\u003e\n        \u003cimg src='https://readthedocs.org/projects/braintaichi/badge/?version=latest' alt='Documentation Status' /\u003e\n    \u003c/a\u003e  \t\n    \u003ca href=\"https://badge.fury.io/py/braintaichi\"\u003e\u003cimg alt=\"PyPI version\" src=\"https://badge.fury.io/py/braintaichi.svg\"\u003e\u003c/a\u003e\n    \u003ca href=\"https://github.com/chaoming0625/braintaichi/actions/workflows/CI.yml\"\u003e\u003cimg alt=\"Continuous Integration\" src=\"https://github.com/chaoming0625/braintaichi/actions/workflows/CI.yml/badge.svg\"\u003e\u003c/a\u003e\n\u003c/p\u003e\n\n\n[``braintaichi``](https://github.com/chaoming0625/braintaichi) leverages Taichi Lang to customize brain dynamics operators.\n\n\n\n## Quick Start\n\n```python\n\n\nimport taichi as ti\n\nimport braintaichi as bti\n\n\n# define the custom kernel\n\n@ti.kernel\ndef transpose_bool_homo_kernel(\n    values: ti.types.ndarray(ndim=1),\n    indices: ti.types.ndarray(ndim=1),\n    indptr: ti.types.ndarray(ndim=1),\n    events: ti.types.ndarray(ndim=1),\n    out: ti.types.ndarray(ndim=1)\n):\n    value = values[0]\n    ti.loop_config(serialize=True)\n    for row_i in range(indptr.shape[0] - 1):\n        if events[row_i]:\n            for j in range(indptr[row_i], indptr[row_i + 1]):\n                out[indices[j]] += value\n\n\nkernel = bti.XLACustomOp(\n    cpu_kernel=transpose_bool_homo_kernel,\n    gpu_kernel=transpose_bool_homo_kernel,\n)\n\n# run with the sample data\n\nimport numpy as np\nimport jax\nimport jax.numpy as jnp\nfrom scipy.sparse import csr_matrix\n\ncsr = csr_matrix((np.random.rand(10, 10) \u003c 0.5).astype(float))\nevents = np.random.rand(10) \u003c 0.5\n\nout = kernel(\n    jnp.array(csr.data),\n    jnp.array(csr.indices),\n    jnp.array(csr.indptr),\n    events,\n    outs=[jax.ShapeDtypeStruct([10], dtype=jnp.float32)]\n)\nprint(out)\n```\n\n\n## Installation\n\n### From PyPI\n\nYou can install ``braintaichi`` via pip:\n\n```bash\npip install braintaichi --upgrade\n```\n\n### From Source\n\nTo compile and install from source code:\n\n#### Prerequisites\n\n1. **Install dependencies**:\n   ```bash\n   pip install taichi==1.7.3 pybind11\n   ```\n\n#### Platform-specific Requirements\n\n**Linux with GPU support**:\n- NVIDIA GPU drivers\n- CUDA Toolkit (version 12.8 recommended, see [ci/linux/gpu/prepare_build_wheel_on_cuda.sh](ci/linux/gpu/prepare_build_wheel_on_cuda.sh))\n- Development packages:\n  ```bash\n  # Example for RHEL/CentOS\n  yum install -y python3-devel cuda-cudart-devel\n  ```\n\n**Windows**:\n- Visual Studio with C++ development tools ([Installation Guide](https://docs.microsoft.com/en-us/cpp/build/vscpp-step-0-installation))\n\n#### Build and Install\n\n```bash\ngit clone https://github.com/chaoming0625/braintaichi.git\ncd braintaichi\npip install -e .\n```\n\nOr alternatively:\n\n```bash\npython setup.py install\n```\n\n## Documentation\n\nThe official documentation is hosted on Read the Docs: [https://braintaichi.readthedocs.io](https://braintaichi.readthedocs.io)\n\n\n## See also the ecosystem\n\nWe are building the brain modeling ecosystem: https://brainmodeling.readthedocs.io/\n\n\n## Citation\n\nIf you think `braintaichi` is significant in your work, please consider to cite the following pubilication:\n\n```bibtex\n\n@inproceedings{wang2024brainpy,\n    title={A differentiable brain simulator bridging brain simulation and brain-inspired computing},\n    author={Wang, Chaoming and Zhang, Tianqiu and He, Sichao and Gu, Hongyaoxing and Li, Shangyang and Wu, Si},\n    booktitle={The Twelfth International Conference on Learning Representations},\n    year={2024}\n}\n```\n\n\n","funding_links":[],"categories":["Libraries"],"sub_categories":["New Libraries"],"project_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fchaobrain%2Fbraintaichi","html_url":"https://awesome.ecosyste.ms/projects/github.com%2Fchaobrain%2Fbraintaichi","lists_url":"https://awesome.ecosyste.ms/api/v1/projects/github.com%2Fchaobrain%2Fbraintaichi/lists"}