zizai

notebooks

AI-Feynman

bullet3

Bullet Physics SDK: real-time collision detection and multi-physics simulation for VR, games, visual effects, robotics, machine learning etc.

ceviche

:shrimp: Electromagnetic Simulation + Automatic Differentiation

chemreps

Implementation of various machine learning representations for molecules

e3nn

ferminet

An implementation of the Fermionic Neural Network for ab-initio electronic structure calculations

gap

Main development repository for GAP - Groups, Algorithms, Programming, a System for Computational Discrete Algebra

jax

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

jax-md

Differentiable, Hardware Accelerated, Molecular Dynamics

lagrangian_nns

Lagrangian Neural Networks

legume

🌱 Guided-mode expansion of photonic crystal slabs

lie_learn

Computations involving Lie groups and harmonic analysis

multi-agent-emergence-environments

Environment generation code for the paper "Emergent Tool Use From Multi-Agent Autocurricula"

openmm

OpenMM is a toolkit for molecular simulation using high performance GPU code.

PhysenNet

Code for physics-enhanced deep neural network (PhysenNet).

pifuhd

High-Resolution 3D Human Digitization from A Single Image.

pymatgen

Python Materials Genomics (pymatgen) is a robust materials analysis code that defines core object representations for structures and molecules with support for many electronic structure codes. It is currently the core analysis code powering the Materials Project.

pymatgen-diffusion

This add-on to pymatgen provides tools for analyzing diffusion in materials.

pytorch_geometric

Geometric Deep Learning Extension Library for PyTorch

rlax

RLMM

RLMM is a reinforcement learning env for molecular modeling (currently only protein-ligand docking).

rlpyt

Reinforcement Learning in PyTorch

RoboGrammar

RoboGrammar: Graph Grammar for Terrain-Optimized Robot Design (SIGGRAPH Asia 2020)

spins-b

Photonic optimization library

symbolic_deep_learning

Code for "Discovering Symbolic Models from Deep Learning with Inductive Biases"

taichi

Productive & portable programming language for high-performance, sparse & differentiable computing

usv_sim_lsa

Unmanned Surface Vehicle simulation on Gazebo with water current and winds

econ-research

neural-generators