JiahaoYao

awesome-ray

Ray - A curated list of resources: https://github.com/ray-project/ray

brc-slurm

This is the manual for the head start at berkeley research computing cluster

ray-jax-tpu-pod-demos

Demos starting ray cluster on tpu pod

JiahaoYao

alpa

Auto parallelization for large-scale neural networks

Cirq

A python framework for creating, editing, and invoking Noisy Intermediate Scale Quantum (NISQ) circuits.

ray

A system for parallel and distributed Python that unifies the ML ecosystem.

dash-extension

compiled dash

dash_cytoscape

dash cytoscape

dash_interactive_graphviz

dash interactive graphviz

Dreambooth-Stable-Diffusion

Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion

flax

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

jax

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

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.

mesa-safe-rl

neural_stpp

Deep generative modeling for time-stamped heterogeneous data, enabling high-fidelity models for a large variety of spatio-temporal domains.

progen

Implementation and replication of ProGen, Language Modeling for Protein Generation, in Jax

qiskit-aer

Aer is a high performance simulator for quantum circuits that includes noise models

qiskit-terra

Qiskit is an open-source SDK for working with quantum computers at the level of extended quantum circuits, operators, and algorithms.

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.

QuSpin

A python wrapper for doing ED calculations on many-body systems

ray-on-aml

Turning AML compute into Ray cluster

ray_lightning

Pytorch Lightning Distributed Accelerators using Ray

se3-transformer-pytorch

Implementation of SE3-Transformers for Equivariant Self-Attention, in Pytorch. This specific repository is geared towards integration with eventual Alphafold2 replication.

SMURF

tf2jax

torchdrug

A powerful and flexible machine learning platform for drug discovery

training_results_v0.7

Training v0.7 results

UCB_math54_2022Fall

Math 54 Section (Fall 2022)

vmcnet

Flexible, general-purpose VMC framework, built on JAX.