balthazarneveu / mva_pepites

Related projects: ⓘ

• erachelson / RLclass_MVA
  ☆31Updated this week
• BlackHC / neural_net_checklist
  ☆137Updated last month
• TorchJD / torchjd
  Library for Jacobian descent with PyTorch. It enables optimization of neural networks with multiple losses (e.g. multi-task learning).
  ☆126Updated this week
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆155Updated 7 months ago
• nanowell / AdEMAMix-Optimizer-Pytorch
  The AdEMAMix Optimizer: Better, Faster, Older.
  ☆132Updated last week
• johnmarktaylor91 / torchlens
  Package for extracting and mapping the results of every single tensor operation in a PyTorch model in one line of code.
  ☆456Updated 3 weeks ago
• lightning-uq-box / lightning-uq-box
  Lightning-UQ-Box: Uncertainty Quantification for Neural Networks with PyTorch and Lightning
  ☆124Updated last month
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆639Updated 4 months ago
• PeaBrane / mamba-tiny
  Simple, minimal implementation of the Mamba SSM in one pytorch file. More efficient than using for loops, but probably less efficient tha…
  ☆89Updated 5 months ago
• SynodicMonth / ChebyKAN
  Kolmogorov-Arnold Networks (KAN) using Chebyshev polynomials instead of B-splines.
  ☆336Updated 4 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆316Updated last month
• CLAIRE-Labo / python-ml-research-template
  A template for starting reproducible Python machine-learning projects with hardware acceleration. Find an example at https://github.com/C…
  ☆57Updated last week
• giakoumoglou / pyssl
  Self-Supervised Learning in PyTorch
  ☆126Updated 6 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆445Updated 6 months ago
• xl0 / lovely-tensors
  Tensors, for human consumption
  ☆1,096Updated 2 weeks ago
• tintn / vision-transformer-from-scratch
  A Simplified PyTorch Implementation of Vision Transformer (ViT)
  ☆123Updated 3 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆246Updated 3 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆236Updated 8 months ago
• KindXiaoming / BIMT
  Brain-Inspired Modular Training (BIMT), a method for making neural networks more modular and interpretable.
  ☆159Updated last year
• Indoxer / LKAN
  Variations of Kolmogorov-Arnold Networks
  ☆112Updated 4 months ago
• jonbarron / tabilize
  Simple code for generating a color-coded latex table from raw data
  ☆146Updated 10 months ago
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆63Updated 5 months ago
• kach / gradient-descent-the-ultimate-optimizer
  Code for our NeurIPS 2022 paper
  ☆360Updated last year
• HHU-MMBS / RepresentationLearning_SS2023
  Representation Learning MSc course Summer Semester 2023
  ☆68Updated last year
• lindermanlab / S5
  ☆248Updated last month
• ironjr / grokfast
  Official repository for the paper "Grokfast: Accelerated Grokking by Amplifying Slow Gradients"
  ☆488Updated 2 months ago
• epistoteles / TensorHue
  TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…
  ☆102Updated last week
• google-deepmind / uncertain_ground_truth
  Dermatology ddx dataset, Jax implementations of Monte Carlo conformal prediction, plausibility regions and statistical annotation aggrega…
  ☆534Updated 5 months ago
• M2Lschool / tutorials2022
  The M2L school 2022 tutorials
  ☆35Updated 2 years ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆118Updated 9 months ago