AI/CV

ChoreoAI

ChoreoAI @ GSoC 2024 Code | Blog Post

A Probability-and-Attention-based Variational Autoencoder model for duet generation. For a hands-on experience, check out this Colab Notebook.

Event Stereo

Event Stereo Code | Paper

EventStereo is a SNN-based model for estimating depth from the scenario using asynchronous stereo events, which can reduce 20x FLOPs when compared to CNN model.

Second-Order Optimizer

Second-order Optimizer Code | Report

A practical second-order optimizer (Gauss-Newton method) made for Deep Learning models using JAX. Damping method, Hessian-Vector Product and Conjugate Gradient techniques are utilized.