wls2002
|
01b7731231
|
optimize import
|
2023-06-29 09:41:49 +08:00 |
|
wls2002
|
114ff2b0cc
|
modifying
|
2023-06-27 18:47:47 +08:00 |
|
wls2002
|
ba369db0b2
|
Perfect!
Next is to connect with Evox!
|
2023-06-25 02:57:45 +08:00 |
|
wls2002
|
0cb2f9473d
|
finish ask part of the algorithm;
use jax.lax.while_loop in graph algorithms and forward function;
fix "enabled not care" bug in forward
|
2023-06-25 00:26:52 +08:00 |
|
wls2002
|
5cbe3c14bb
|
Current Progress: After final design presentation
|
2023-06-19 15:17:56 +08:00 |
|
wls2002
|
acedd67617
|
change project structure and using .ini as config file
|
2023-06-15 11:05:26 +08:00 |
|
wls2002
|
bd421de9ad
|
Current Progress: After final design presentation
|
2023-06-14 10:20:55 +08:00 |
|
wls2002
|
72c9d4167a
|
FAST!
|
2023-05-13 20:58:03 +08:00 |
|
wls2002
|
6006f92f3f
|
finish jit-able speciate function
next time i'll create a new branch
|
2023-05-12 19:26:02 +08:00 |
|
wls2002
|
9b56f4ff73
|
做了一些时间测试：
1. jit中的vmap一个函数不会触发重新编译
2. jax.lax.while_lop单独执行确实可以提前中断，但在vmap中的性能需要考虑。
|
2023-05-12 16:42:57 +08:00 |
|
wls2002
|
47b1a1dbb2
|
refactor genome.py use (C, 4) to replace (2, N, N) to represent connections
faster, faster and faster!
|
2023-05-12 00:57:55 +08:00 |
|
wls2002
|
e5fc1167d9
|
refactor genome.py use (C, 4) to replace (2, N, N) to represent connections
use "cons" to replace "connections" in code for beauty
|
2023-05-11 19:49:19 +08:00 |
|
wls2002
|
e2a5117554
|
initial commit in branch developing
|
2023-05-11 19:04:56 +08:00 |
|
wls2002
|
dde338696f
|
reduce the use of device_get in speciate
|
2023-05-08 15:58:09 +08:00 |
|
wls2002
|
d4a75b9394
|
use jit().lower.compile in create functions
|
2023-05-08 02:35:04 +08:00 |
|
wls2002
|
497d89fc69
|
move o2o_distance and o2m_distance to pipelines
|
2023-05-08 01:19:45 +08:00 |
|
wls2002
|
73ac1bcfe0
|
generally complete, but not work well. Debug
|
2023-05-06 11:35:44 +08:00 |
|
wls2002
|
6faa07f507
|
initial commit
|
2023-05-05 14:19:13 +08:00 |
|