tisabe

Tim Bechtel's starred repositories

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

Language:PythonApache-2.06010 86 899

A Graph Neural Network Library in Jax

Language:PythonApache-2.01359 39 38

ORB forcefield models from Orbital Materials

Language:PythonNOASSERTION157 4 9

Various code/notebooks to benchmark different ways we could estimate uncertainty in ML predictions.

Language:Jupyter Notebook39 60