Narges Javaheri's starred repositories
get-started-with-JAX
The purpose of this repo is to make it easy to get started with JAX, Flax, and Haiku. It contains my "Machine Learning with JAX" series of tutorials (YouTube videos and Jupyter Notebooks) as well as the content I found useful while learning about the JAX ecosystem.
bayesian-algorithm-execution
Bayesian algorithm execution (BAX)
jammy_flows
A package to describe amortized (conditional) normalizing-flow PDFs defined jointly on tensor products of manifolds with coverage control. The connection between different manifolds is fixed via an autoregressive structure.
pytorch-iobs
A lightweight implementation of Information-Ordered Bottlenecks (IOBs) in PyTorch.
cosmopower
Machine Learning - accelerated Bayesian inference
cvxpylayers
Differentiable convex optimization layers
complexPyTorch
A high-level toolbox for using complex valued neural networks in PyTorch
modulus-makani
Massively parallel training of machine-learning based weather and climate models
Computational_Physics_in_Python
A repository of codes for computational physics in Python.
ComputationalPhysics300
computational physics class taught at UNLV (Phys300)
deepmind-research
This repository contains implementations and illustrative code to accompany DeepMind publications
modulus-sym
Framework providing pythonic APIs, algorithms and utilities to be used with Modulus core to physics inform model training as well as higher level abstraction for domain experts