Pinned

1. Recurrent-Linear-Transformers Recurrent-Linear-Transformers Public

   Implementation of Recurrent Linear Transformers in Jax+Flax.

   Python 12 1

2. OmniNet OmniNet Public

   Official Pytorch implementation of "OmniNet: A unified architecture for multi-modal multi-task learning" | Authors: Subhojeet Pramanik, Priyanka Agrawal, Aman Hussain

   Python 509 60

3. pytorch2jax pytorch2jax Public

   Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.

   Python 18

4. Infinite-Runner-Ultimate Infinite-Runner-Ultimate Public

   A cross-platform procedural environment generation and Endless Runner engine. Features real-time object pooling and includes an interactive level editor.

   C# 83 18

5. Recurrent-PPO-Jax Recurrent-PPO-Jax Public

   Implementation of Proximal Policy Optimization in Jax+Flax

   Python 13 1

6. cognibit/Text-Normalization-Demo cognibit/Text-Normalization-Demo Public

   Demonstration of the results in "Text Normalization using Memory Augmented Neural Networks", Authors: Subhojeet Pramanik, Aman Hussain

   Jupyter Notebook 59 7