Arjun Parthasarathy

icon location Palo Alto, CA icon location I.I Rabi Scholar, CS + Applied Math @ Columbia

Results 5 issues of Arjun Parthasarathy

Python wheel for linux/arm64

Hi, Would it be possible for you to create a python wheel for linux arm64 or aarch64? I'm trying to create a custom wheel but I'm not able to. If...

BERT MLM Support

1 comment

## Description @AlanAboudib this adds support for a BERT encoder and iterator, specifically for the Masked LM use case. ## Affected Dependencies Now requires [HuggingFace Transformers](https://github.com/huggingface/transformers) library to be installed....

256 Kernel

16x16 kernel using extension of producer-consumer template

Refactor complex code into separate files + add old kernel versions

Trajectory Replay in DMControl environment doesn't match dataset

Hi, I've been trying to replay the trajectories in the TDMPC2 dataset for the Cartpole Swingup DM Control Suite task, and I find that when I reset the environment to...