Name: Jiaxin Zhang
Type: User
Company: Intuit AI Research
Bio: AI Research Scientist on LLM reliability, evaluation, fine-tuning, and optimization
Twitter: jxzhangjhu
Location: Mountain View
Blog: https://jxzhangjhu.github.io
Jiaxin Zhang's Projects
Code to reproduce results in "Preconditioned training of normalizing flows for variational inference in inverse problems"
A large-scale dataset of both raw MRI measurements and clinical MRI images
Try several methods for MRI reconstruction on the fastmri dataset. Home to the XPDNet, runner-up of the 2020 fastMRI challenge.
Repository for the Explainable Deep One-Class Classification paper
A collection of Google research projects related to Federated Learning and Federated Analytics.
FedNLP: A Research Platform for Federated Learning in Natural Language Processing. Accepted to NAACL 2022
Implementation of Few-shot Domain Adaptation by Causal Mechanism Transfer (ICML 2020)
[CoRL 2020] Fit2Form: 3D Generative Model for Robot Gripper Form Design
Official Pytorch Implementation for "Fix the Noise: Disentangling Source Feature for Transfer Learning of StyleGAN" (CVPRW 2022 Best paper)
An open-source Python package for creating fast and accurate interatomic potentials.
An Open Flexible Quadrotor Simulator
Implicit Normalizing Flows + Reinforcement Learning
This library would form a permanent home for reusable components for deep probabilistic programming. The library would form and harness a community of users and contributors by focusing initially on complete infra and documentation for how to use and create components.
Flowtron is an auto-regressive flow-based generative network for text to speech synthesis with control over speech variation and style transfer
A Library for Uncertainty Quantification.
Framework for Easily Invertible Architectures
G-SchNet - a generative model for 3d molecular structures
GAAL-based Outlier Detection
Graph Auto-Encoder in PyTorch
Tooling for GANs in TensorFlow
Offical implementation of "Graph-Augmented Normalizing Flows for Anomaly Detection of Multiple Time Series" (ICLR 2022)
A toolkit for reproducible reinforcement learning research.
Implementation of Graph Convolutional Networks in TensorFlow
Official Code Repository for the paper "Score-based Generative Modeling of Graphs via the System of Stochastic Differential Equations" (ICML 2022)
A general-purpose probabilistic programming system with programmable inference