ameya98

ActionAngleNetworks

The official JAX implementation of Action-Angle Networks!

DeepGenerativeModels

PyTorch Implementations of Popular Deep Generative Models.

LearningForceFields

Learning Molecular Force Fields with e3nn

PolymorphismInC

An example of how to emulate runtime polymorphism in C via function-pointers.

pr2roc

Resample precision-recall curves correctly!

saxpy

SAX (Symbolic Aggregate approXimation) and algorithms (HOT-SAX and SAX-VSM) Python implementations

snap-python

SNAP Python code, SWIG related files

allegro-jax

ameya98.github.io

Just a plain, simple and elegant one-page theme for research/academia.

CommonLoopUtils

My fork of CLU for MacOS-compatibility.

DensityRatioEstimation

A Python Package for Density Ratio Estimation

DisCOVID

Team 88's repository for the JHU Design Challenge.

e3nn-jax

Ameya's fork of e3nn-jax.

electron-chat-app-demo

Bootleg WhatsApp Chat with Electron and React

flax

Flax is a neural network library for JAX that is designed for flexibility.

GNNs-Recipe

A recipe to study Graph Neural Networks (GNNs)

iclr-blog-track.github.io

Ameya's fork of the blog post.

matrixprofile-ts

A Python library for detecting patterns and anomalies in massive datasets using the Matrix Profile

nequip-jax

ns2.35

ns2 for research

schnetpack

SchNetPack - Deep Neural Networks for Atomistic Systems

schnetpack-gschnet

G-SchNet extension for SchNetPack

DIG

A library for graph deep learning research

e3_diffusion_for_molecules

Ameya's fork of EDM

google-research

Google Research

jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

minified-generative-models

Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs.

NTK-Generalization

Reproducing the results of 'Spectral Bias and Task-Model Alignment Explain Generalization in Kernel Regression and Infinitely Wide Neural Networks'

pyspectra

Library for computing, inverting, and visualizing spectra of local atomic environments

st-speckmol

🧬Streamlit Component for creating Speck molecular structures within Streamlit Web app🚀.