farenga's Stars
google/flax
Flax is a neural network library for JAX that is designed for flexibility.
ENSTA-U2IS-AI/torch-uncertainty
Open-source framework for uncertainty and deep learning models in PyTorch :seedling:
CUQI-DTU/CUQIpy-FEniCS
CUQIpy plug-in for FEniCS PDE modeling
jayjay-park/Diff_MultiPhysics
Reference-LAPACK/lapack
LAPACK development repository
openxla/xla
A machine learning compiler for GPUs, CPUs, and ML accelerators
jiaweizzhao/GaLore
GaLore: Memory-Efficient LLM Training by Gradient Low-Rank Projection
williamgilpin/cphy
Class materials for computational physics course
dblalock/cuda_ext_example
Simple example project that builds a PyTorch CUDA extension
kth-competitive-programming/kactl
KTH Algorithm Competition Template Library (... eller KTHs AC-tillverkande lapp)
NVIDIA/cutlass
CUDA Templates for Linear Algebra Subroutines
hollance/neural-engine
Everything we actually know about the Apple Neural Engine (ANE)
karpathy/nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
corsix/amx
Apple AMX Instruction Set
aorvieto/noise_injection_overparam
Code for "Explicit Regularization in Overparametrized Models via Noise Injection"
google-research/tuning_playbook
A playbook for systematically maximizing the performance of deep learning models.
microsoft/torchscale
Foundation Architecture for (M)LLMs
Koopman-Laboratory/KoopmanLab
A library for Koopman Neural Operator with Pytorch.
mert-kurttutan/torchview
torchview: visualize pytorch models
rrwabina/SDENet-UQ-ESL
SDENet as Uncertainty Quantification Method for EEG Source Localization
theislab/PathReg
Sparsity-enforcing regularizer
twitter-research/graph-neural-pde
Graph Neural PDEs
martenlienen/torchode
A parallel ODE solver for PyTorch
barkm/torch-fenics
PyTorch-FEniCS interface
pdebench/PDEBench
PDEBench: An Extensive Benchmark for Scientific Machine Learning
google-deepmind/educational
patrick-kidger/torchtyping
Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.
huggingface/accelerate
🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
lucidrains/denoising-diffusion-pytorch
Implementation of Denoising Diffusion Probabilistic Model in Pytorch
google/snappy
A fast compressor/decompressor