niklasschmitz's Stars
huggingface/transformers
🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
CompVis/stable-diffusion
A latent text-to-image diffusion model
karpathy/nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
NVlabs/instant-ngp
Instant neural graphics primitives: lightning fast NeRF and more
google-research/arxiv-latex-cleaner
arXiv LaTeX Cleaner: Easily clean the LaTeX code of your paper to submit to arXiv
google-deepmind/alphatensor
alecjacobson/common-3d-test-models
Repository containing common 3D test models in original format with original source if known and obj mesh
patrick-kidger/jaxtyping
Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/
google/learned_optimization
probml/dynamax
State Space Models library in JAX
RobertTLange/gymnax
RL Environments in JAX 🌍
ott-jax/ott
Optimal transport tools implemented with the JAX framework, to get differentiable, parallel and jit-able computations.
JuliaDebug/Infiltrator.jl
No-overhead breakpoints in Julia
lucidrains/equiformer-pytorch
Implementation of the Equiformer, SE3/E3 equivariant attention network that reaches new SOTA, and adopted for use by EquiFold for protein folding
pdearena/pdearena
chr5tphr/zennit
Zennit is a high-level framework in Python using PyTorch for explaining/exploring neural networks using attribution methods like LRP.
GPflow/GeometricKernels
Geometric kernels on manifolds, meshes and graphs
nmwsharp/neural-implicit-queries
Queries on neural implicit surfaces via range analysis: ray casting, intersection, closest point, & more. SIGGRAPH 2022 paper. JAX implementation.
ViCCo-Group/thingsvision
Python package for extracting representations from state-of-the-art computer vision models
stefanch/sGDML
sGDML - Reference implementation of the Symmetric Gradient Domain Machine Learning model
thorben-frank/mlff
Build neural networks for machine learning force fields with JAX
NeilGirdhar/tjax
Tools for JAX
adrhill/BrowserMacros.jl
Julia macros to surf the web from the comfort of your REPL. 🏄
mohamed82008/DifferentiableFactorizations.jl
Differentiable matrix factorizations using ImplicitDifferentiation.jl.
ZIB-IOL/Boscia.jl
Mixed-Integer Convex Programming: Branch-and-bound with Frank-Wolfe-based convex relaxations
ml4physics/cyjax
Machine learning Calabi-Yau metrics with JAX
madjax-hep/madjax
madjax
awf/awf-jaxutils
A few utils for JAX (and general python) programming
stratisMarkou/random-walks
My code repo for random-walks.org
stratisMarkou/check-shape
A simple tool for checking array and tensor shapes