Alternatives and similar repositories for ezpz

Users that are interested in ezpz are comparing it to the libraries listed below

• argonne-lcf / inference-endpoints
  This is a repository with examples to run inference endpoints on various ALCF clusters
  ☆24Updated 3 weeks ago
• IntelLabs / EquiTriton
  EquiTriton is a project that seeks to implement high-performance kernels for commonly used building blocks in equivariant neural networks…
  ☆62Updated last week
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆113Updated 8 months ago
• lucidrains / triangle-multiplicative-module
  Implementation of the Triangle Multiplicative module, used in Alphafold2 as an efficient way to mix rows or columns of a 2d feature map, …
  ☆35Updated 4 years ago
• maxxxzdn / erwin
  Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]
  ☆99Updated 2 months ago
• argonne-lcf / Megatron-DeepSpeed
  Ongoing research training transformer language models at scale, including: BERT & GPT-2
  ☆16Updated this week
• NERSC / slurm-ray-cluster
  ☆59Updated 2 years ago
• ORNL / HydraGNN
  Distributed PyTorch implementation of multi-headed graph convolutional neural networks
  ☆85Updated this week
• fishmingyu / GeoT
  GeoT: Tensor Centric Library for Graph Neural Network via Efficient Segment Reduction on GPU
  ☆22Updated 5 months ago
• NERSC / sc22-dl-tutorial
  Material for the SC22 Deep Learning at Scale Tutorial
  ☆41Updated 2 years ago
• argonne-lcf / ALCFBeginnersGuide
  ☆43Updated last month
• NERSC / nersc-dl-wandb
  Guidelines on using Weights and Biases logging for deep learning applications on NERSC machines
  ☆13Updated 2 years ago
• Lemon-cmd / energy-transformer-graph
  This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…
  ☆25Updated last year
• KernelTuner / kernel_tuner_tutorial
  A hands-on introduction to tuning GPU kernels using Kernel Tuner https://github.com/KernelTuner/kernel_tuner/
  ☆32Updated 4 months ago
• google-research / spherical-cnn
  ☆125Updated 8 months ago
• axonn-ai / axonn
  A parallel framework for training deep neural networks
  ☆63Updated 5 months ago
• divelab / MFA
  Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.
  ☆17Updated 7 months ago
• ramanathanlab / genslm
  GenSLMs: Genome-scale language models reveal SARS-CoV-2 evolutionary dynamics
  ☆134Updated last year
• NERSC / sc23-dl-tutorial
  SC23 Deep Learning at Scale Tutorial Material
  ☆47Updated 11 months ago
• stfc-sciml / sciml-bench
  SciML Benchmarking Suite for AI for Science
  ☆42Updated last year
• graphcore / ogb-lsc-pcqm4mv2
  The OGB-LSC is the Large Scale Competition by Open Graph Benchmark to help accelerate research into machine learning on graph structured …
  ☆79Updated last year
• Graph-COM / HEPT
  [ICML 2024 Oral] LSH-Based Efficient Point Transformer (HEPT)
  ☆21Updated 7 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆133Updated last year
• kabouzeid / turm
  TUI for the Slurm Workload Manager
  ☆192Updated this week
• pgniewko / forward_forward_vhts
  The Forward-Forward Algorithm for Drug Discovery
  ☆34Updated 2 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• argonne-lcf / ai-science-training-series
  ☆224Updated 9 months ago
• pyg-team / pyg-lib
  Low-Level Graph Neural Network Operators for PyG
  ☆210Updated this week
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆42Updated 8 months ago