Differentiable ODE solvers with full GPU support and O(1)-memory backpropagation.

Body Transformer: Leveraging Robot Embodiment for Policy Learning

A library to generate LaTeX expression from Python code.

Pytorch implementation of convolutional neural network visualization techniques

Lie Algebra Convolutional Network implementation

Step by step explanation of 2D convolution implemented as matrix multiplication using toeplitz matrices

Code for Max F. Burg et al. (2021): Learning Divisive Normalization in Primary Visual Cortex

Democratizing Reinforcement Learning for LLMs

Implementation of Convolutional LSTM in PyTorch.

Comprehensive Implementation of Proximal Policy Optimization

Implementation of "Visual number sense in untrained deep neural networks" (Kim et al., Science Advances, 2021)

High-quality single file implementation of Deep Reinforcement Learning algorithms with research-friendly features (PPO, DQN, C51, DDPG, TD3, SAC, PPG)

A library for advanced large language model reasoning

Cohere Toolkit is a collection of prebuilt components enabling users to quickly build and deploy RAG applications.

Official PyTorch implementation of the Fishr regularization for out-of-distribution generalization

Machine Learning Interviews from FAANG, Snapchat, LinkedIn. I have offers from Snapchat, Coupang, Stitchfix etc. Blog: mlengineer.io.

Neural Logic Reasoning for Question Answering

Jupyter notebooks and markdownfiles that serve as tutorials for the packages of JuliaDynamics

Computing on Topological Domains

NYU PSYCH-GA 3405.001 / DS-GA 3001.014 : Advancing AI through cognitive science

Author implementation of the paper "CommonsenseQA: A Question Answering Challenge Targeting Commonsense Knowledge"

Ollama Python library

LLM inference in C/C++

Instruct-tune LLaMA on consumer hardware

Code and documentation to train Stanford's Alpaca models, and generate the data.

ChristGPT is a Large Language Model trained on the Bible and Jesus's teachings

Prod Env