Rohan Potdar's Projects
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
Ray Aviary - evaluate multiple LLMs easily
Go ahead and axolotl questions
A recommender system to find related courses based on course descriptions and NLP
Implentation of COMA using RLlib.
Combined Model-Based and Inverse Reinforcement Learning for Generalization
Learning Invariant Representations for Reinforcement Learning without Reconstruction
Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.
Implementation of Dreamer v3 in pytorch.
Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.
A collection of useful .gitignore templates
A toolkit for developing and comparing reinforcement learning algorithms.
Main repo for JOS labs
A customisable 2D platform for agent-based AI research
⚡ Building applications with LLMs through composability ⚡
Multi-Agent Reinforcement Learning with Stable-Baselines3
Library for Model Based RL
🚀 Accelerate training and inference of 🤗 Transformers and 🤗 Diffusers with easy to use hardware optimization tools
Gym for multi-agent reinforcement learning
Training code PRIMAL2 - Public Repo
Tensors and Dynamic neural networks in Python with strong GPU acceleration
An open source framework that provides a simple, universal API for building distributed applications. Ray is packaged with RLlib, a scalable reinforcement learning library, and Tune, a scalable hyperparameter tuning library.
A training framework for Stable Baselines3 reinforcement learning agents, with hyperparameter optimization and pre-trained agents included.
Extra tools for RLGym, like sb3 compatibility