Chaoming Wang

Please: - [x] Check for duplicate requests. - [x] Describe your goal, and if possible provide a code snippet with a motivating example. Currently, the ``brainpy.dyn.DSRunner`` and ``brainpy.integrators.IntegratorRunner`` have good...

Support veterization and parallerization for class objects

1

Please: - [x] Check for duplicate requests. - [x] Describe your goal, and if possible provide a code snippet with a motivating example. If instances of ``brainpy.math.Variable`` in a class...

Please: - [x] Check for duplicate requests. - [x] Describe your goal, and if possible provide a code snippet with a motivating example. Rate-based whole brain modeling is attractive for...

Cannot build the project

1

Dear @yzhaiustc , Thanks for you amazing effort for this repo. However, we cannot build this project, because the ``helper_functions..h`` is missing. Can you provide such files? Thanks!

Dear all, I want to use Triton to implement a matrix multiplication which computes $OUT = S * A * C$, where $S \in \mathbb{R}^{m}$, $A \in \mathbb{R}^{m * k}$,...

Currently, we only see the support for warp-level primitives for 32-bit data, such as ``ti.simt.warp.shfl_sync_f32``, ``ti.simt.warp.shfl_up_f32``, ``ti.simt.warp.shfl_down_f32``. These primitives cannot be applied to ``f64``. We are expecting support for float64...

Fusing Taichi with JAX

12

We have already seen some examples which can use Taichi as a part of the PyTorch program. For example, - https://github.com/ailzhang/blog_code/blob/master/rwkv/benchmark.py - https://github.com/ifsheldon/stannum However, is it possible to integrate Taichi...

- rename `clean_caches` to `clear_taichi_aot_caches` - rename `check_kernels_count` to `count_taichi_aot_kernels` - support to clean taichi caches in `brainpy.math.clear_taichi_aot_caches`