Sun-Xiaohui

How does taskflow schedule multiple tasks on an Executor with multiple workers?

1

I plan to schedule tasks with taskflow in a real-time system, so I need to know how long it takes to finish a task and make sure every task meets...

gpu任务能耗相关

1

老师您好，我现在有一个周期性的gpu任务，但是我用nvidia-smi工具发现，在执行完一次任务等待第二次执行该任务的间隔内，gpu功耗并没有降到静态功耗，这是什么原因，该如何解决呢？我的gpu静态功耗是29W，执行任务时功耗会升至120W，执行完功耗就保持87W左右，再执行任务又升至120W。

Add CRF module

1

Conditional Random Field (CRF) is a basic model of natural language processing, which is widely used in word segmentation, named entity recognition, part-of-speech tagging and other tagging scenarios. For example,...

jax.experimental.rnn.lstm() violates JIT invariance

9

``` import jax import jax.numpy as jnp from jax.experimental import rnn batch_size = 8 input_size = 8 hidden_size = 4 num_layers = 1 bidirectional = True num_directions = 2 if...

More memory consume compared with Pytorch

2

Hello, flax team. When I tried to transfer Pytorch's models to the flax framework, I find flax will consume more memory than Pytorch's. For example, a ResNet50 model in Pytorch...