choltz95's Stars
tinygrad/tinygrad
You like pytorch? You like micrograd? You love tinygrad! ❤️
google-deepmind/optax
Optax is a gradient processing and optimization library for JAX.
Liuhong99/Sophia
The official implementation of “Sophia: A Scalable Stochastic Second-order Optimizer for Language Model Pre-training”
blackjax-devs/blackjax
BlackJAX is a Bayesian Inference library designed for ease of use, speed and modularity.
google/jax-cfd
Computational Fluid Dynamics in JAX
pyamg/pyamg
Algebraic Multigrid Solvers in Python
patrick-kidger/lineax
Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax
circuitnet/CircuitNet
CircuitNet: An Open-Source Dataset for Machine Learning Applications in Electronic Design Automation (EDA)
TILOS-AI-Institute/MacroPlacement
Macro Placement - benchmarks, evaluators, and reproducible results from leading methods in open source
albarji/proxTV
Matlab and Python toolbox for fast Total Variation proximity operators
danspielman/Laplacians.jl
Algorithms inspired by graph Laplacians: linear equation solvers, sparsification, clustering, optimization, etc.
leowyy/GraphTSNE
PyTorch Implementation of GraphTSNE, ICLR’19
stevecheckoway/ucsddissertation
LaTeX class file for writing dissertations at UC San Diego
nightldj/admm_release
Matlab/Python code for the ADMM part of my thesis ''Alternating Optimization: Constrained Problems, Adversarial Networks, and Robust Models''
vlad17/np-learn
Jupyter notebook-post of advanced numpy techniques
pdobsan/pynauty
Isomorphism testing and automorphisms of graphs
Farama-Foundation/Jumpy
On-the-fly conversions between Jax and NumPy tensors
alexOarga/haiku-geometric
A collection of graph neural networks implementations in JAX
pyamg/pyamg-examples
Examples in PyAMG
chiggum/pyLDLE2
Implementation of Low Distortion Local Eigenmaps and several variations of it
fast-algos/pIRLS
Fast IRLS code for solving p-norm regression problems
LEAP-WS/CGPN
This is the repository for the NeurIPS-21 paper [Contrastive Graph Poisson Networks: Semi-Supervised Learning with Extremely Limited Labels].
ZhiminPeng/Jacobi-ADMM
Parallel Multi-Block ADMM with o(1/k) Convergence
fast-algos/graphs-matrices-optimization
Lecture notes for CSC2421
qiyang-ustc/jaxeigs
Rescure useful code in TensorNetwork by google for further personal usage.
proloyd/fastapy
FATSA + py - Python implementation of fast adaptive proximal gradient descent algorithm
jackd/jju
Jack's Jax Utilities
iggallagher/Spectral-Embedding
oxfordcontrol/DoublyStochastic.jl
Doubly Stochastic Approximation of a Matrix in the Frobenius norm.
Dinple/global_routing