lhnguyen102 / cuTAGI

Related projects: ⓘ

• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆103Updated 2 years ago
• google-deepmind / tf2jax
  ☆104Updated last week
• bhoov / hamux
  Hierarchical Associative Memory User Experience
  ☆88Updated 8 months ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆36Updated 5 months ago
• google-deepmind / neural_testbed
  ☆191Updated 4 months ago
• google-deepmind / conformal_training
  This repository contains a Jax implementation of conformal training corresponding to the ICLR'22 paper "learning optimal conformal classi…
  ☆116Updated 2 years ago
• rasbt / cyclemoid-pytorch
  Cyclemoid implementation for PyTorch
  ☆87Updated 2 years ago
• cgarciae / ciclo
  A functional training loops library for JAX
  ☆84Updated 7 months ago
• cgarciae / jax_metrics
  A metrics library for the JAX ecosystem
  ☆36Updated last year
• alonfnt / pcax
  Minimal Principal Component Analysis (PCA) implementation in JAX
  ☆15Updated 6 months ago
• optimizedlearning / mechanic
  ☆35Updated 7 months ago
• jambo6 / neuralRDEs
  Code for: "Neural Rough Differential Equations for Long Time Series", (ICML 2021)
  ☆115Updated 3 years ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆105Updated 4 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆82Updated 2 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (affine group preconditioner, low-rank approximation preconditioner …
  ☆105Updated this week
• pyhopper / pyhopper
  PyHopper is a hyperparameter optimizer, made specifically for high-dimensional problems arising in machine learning research.
  ☆86Updated 8 months ago
• microsoft / bayesianize
  Bayesianize: A Bayesian neural network wrapper in pytorch
  ☆81Updated 4 months ago
• cooper-org / cooper
  A general-purpose, deep learning-first library for constrained optimization in PyTorch
  ☆106Updated this week
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆100Updated 2 months ago
• aradha / recursive_feature_machines
  ☆36Updated last week
• VectorInstitute / vbll
  Simple (and cheap!) neural network uncertainty estimation
  ☆47Updated last week
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆150Updated 2 years ago
• aks2203 / easy-to-hard
  Official repository for the paper "Can You Learn an Algorithm? Generalizing from Easy to Hard Problems with Recurrent Networks"
  ☆58Updated 2 years ago
• nalexai / hyperlib
  Library that contains implementations of machine learning components in the hyperbolic space
  ☆127Updated 5 months ago
• google-research / hyperbo
  Pre-trained Gaussian processes for Bayesian optimization
  ☆78Updated 2 weeks ago
• openopt / chop
  CHOP: An optimization library based on PyTorch, with applications to adversarial examples and structured neural network training.
  ☆76Updated 6 months ago
• davda54 / ada-hessian
  Easy-to-use AdaHessian optimizer (PyTorch)
  ☆77Updated 3 years ago
• mle-infrastructure / mle-hyperopt
  Lightweight Hyperparameter Optimization 🚂
  ☆145Updated 3 weeks ago
• cgarciae / treeo
  A small library for creating and manipulating custom JAX Pytree classes
  ☆59Updated last year
• probabilisticai / probai-2023
  Materials of the Nordic Probabilistic AI School 2023.
  ☆86Updated 11 months ago