Ankit Gupta

tensorflow io on Mac not working with s3 paths

4

I recently installed tensorflow io on my M1 Mac in an arm64 conda environment. the install worked fine, and I'm able to import tensorflow and tensorflow io, but for some...

The existing code defines a variable `precision` ([here](https://github.com/pytorch-labs/gpt-fast/blob/3bcaaaf068d112d534f335ec21a17d7b8b5551bf/generate.py#L266)) which is then used in `_load_model()` [here](https://github.com/pytorch-labs/gpt-fast/blob/3bcaaaf068d112d534f335ec21a17d7b8b5551bf/generate.py#L230) to set the dtype for the model. However, this variable was not getting passed to...

Added a basic Dockerfile to improve dev experience

4

I made a simple docker container (heavily influenced on the official [Tensorflow docker container](https://github.com/tensorflow/build/blob/master/tensorflow_runtime_dockerfiles/gpu.Dockerfile) that I've used for other deep learning projects, and sharing some of the same features as...

need to swap layer norm op for triton-based layer norm?

2

In the Flash-attention repo [here](https://github.com/Dao-AILab/flash-attention/tree/main/csrc/layer_norm), there is now a note that the fused CUDA op has been replaced with a Triton op. in light of that, is it now reasonable...

Resolves https://github.com/HazyResearch/hyena-dna/issues/57