JiahaoYao
Ph.D. @ UCB Math, interested in Computational Mathematics and Deep Learning
University of California, BerkeleyBerkeley, CA
Pinned Repositories
alpa
Auto parallelization for large-scale neural networks
awesome-ray
Ray - A curated list of resources: https://github.com/ray-project/ray
BabyDeepMDpytorch
Light weight version of Deep MD written in pytorch
brc-slurm
This is the manual for the head start at berkeley research computing cluster
ceciliavision
Cirq
A python framework for creating, editing, and invoking Noisy Intermediate Scale Quantum (NISQ) circuits.
DeepStriving
Deep Learning Algorithms in Practice
JiahaoYao
ray
A system for parallel and distributed Python that unifies the ML ecosystem.
ray-jax-tpu-pod-demos
Demos starting ray cluster on tpu pod
JiahaoYao's Repositories
JiahaoYao/awesome-ray
Ray - A curated list of resources: https://github.com/ray-project/ray
JiahaoYao/brc-slurm
This is the manual for the head start at berkeley research computing cluster
JiahaoYao/ray-jax-tpu-pod-demos
Demos starting ray cluster on tpu pod
JiahaoYao/JiahaoYao
JiahaoYao/alpa
Auto parallelization for large-scale neural networks
JiahaoYao/Cirq
A python framework for creating, editing, and invoking Noisy Intermediate Scale Quantum (NISQ) circuits.
JiahaoYao/ray
A system for parallel and distributed Python that unifies the ML ecosystem.
JiahaoYao/dash-extension
compiled dash
JiahaoYao/dash_cytoscape
dash cytoscape
JiahaoYao/deepks-kit
a package for developing machine learning-based chemically accurate energy and density functional models
JiahaoYao/Dreambooth-Stable-Diffusion
Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion
JiahaoYao/flax
Flax is a neural network library for JAX that is designed for flexibility.
JiahaoYao/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
JiahaoYao/lightning
The most intuitive, flexible, way for researchers, ML engineers and data scientists to build models (with PyTorch) and ML systems for the ML lifecycle with an obsessive focus on flexibility and performance.
JiahaoYao/mesa-safe-rl
JiahaoYao/neural_stpp
Deep generative modeling for time-stamped heterogeneous data, enabling high-fidelity models for a large variety of spatio-temporal domains.
JiahaoYao/progen
Implementation and replication of ProGen, Language Modeling for Protein Generation, in Jax
JiahaoYao/qiskit-aer
Aer is a high performance simulator for quantum circuits that includes noise models
JiahaoYao/qiskit-terra
Qiskit is an open-source SDK for working with quantum computers at the level of extended quantum circuits, operators, and algorithms.
JiahaoYao/qlib
Qlib is an AI-oriented quantitative investment platform, which aims to realize the potential, empower the research, and create the value of AI technologies in quantitative investment. With Qlib, you can easily try your ideas to create better Quant investment strategies.
JiahaoYao/QuSpin
A python wrapper for doing ED calculations on many-body systems
JiahaoYao/ray-on-aml
Turning AML compute into Ray cluster
JiahaoYao/ray_lightning
Pytorch Lightning Distributed Accelerators using Ray
JiahaoYao/se3-transformer-pytorch
Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration with eventual Alphafold2 replication.
JiahaoYao/SMURF
JiahaoYao/tf2jax
JiahaoYao/torchdrug
A powerful and flexible machine learning platform for drug discovery
JiahaoYao/training_results_v0.7
Training v0.7 results
JiahaoYao/UCB_math54_2022Fall
Math 54 Section (Fall 2022)
JiahaoYao/vmcnet
Flexible, general-purpose VMC framework, built on JAX.