Software

JAXGenesis

Implementations of different generative model architectures in JAX framework using Equinox JAX Library for Neural Network Implementations. Implemented Generative Model Architectures:

Generative Adversarial Networks, Variational AutoEncoders, Flow-Based Models, Energy-Based Models, NeuralSDEs etc.

SpinningUp RL JAX

Implementations Reinforcement Learning Algorithms in Jax Framework inspired from the DeepRL learning resource “Spinning Up in DeepRL by OpenAI”. Algorithms Implemented in the library: Vanilla Policy Gradient (VPG), Trust Region Policy Optimization (TRPO), Proximal Policy Optimization (PPO), Deep Deterministic Policy Gradient (DDPG), Twin Delayed DDPG (TD3), Soft Actor-Critic (SAC)

MLAlgorithmsKit: A Light-weight Machine Learning Library in Python

Implementation of standard ML Algorithms from scratch using Numpy and Pandas libraries.
Algorithms Implemented:

Linear Regression, Logistic Regression, Multi-Layered Perceptron, K-Means Clustering, KNN (K-Nearest Neighbors), Decision tree