nimahsn

nimahsn's Stars

• jax-ml/jax

  Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

  Language:Python31.8k3286.1k3k

• electronicarts/CnC_Remastered_Collection

  Command & Conquer: Remastered Collection

  Language:C++21k54705.4k

• alshedivat/al-folio

  A beautiful, simple, clean, and responsive Jekyll theme for academics

  Language:HTML12.6k2762311.7k

• gpu-mode/lectures

  Material for gpu-mode lectures

  Language:Jupyter Notebook4.2k6210421

• google/brax

  Massively parallel rigidbody physics simulation on accelerator hardware.

  Language:Jupyter Notebook2.6k34375279

• BasixKOR/awesome-activitypub

  Awesome list of ActivityPub based projects

  1.1k371645

• neuralgcm/neuralgcm

  Hybrid ML + physics model of the Earth's atmosphere

  Language:Python802147499

• probml/dynamax

  A Python package for probabilistic state space modeling with JAX

  Language:Python7932520493

• NVIDIA/torch-harmonics

  Differentiable signal processing on the sphere for PyTorch

  Language:Jupyter Notebook4468540

• Mohamedelrefaie/DrivAerNet

  A Large-Scale Multimodal Car Dataset with Computational Fluid Dynamics Simulations and Deep Learning Benchmarks

  Language:Python291183042

• brentyi/jaxlie

  Rigid transforms + Lie groups for JAX

  Language:Python24761515

• samuela/torch2jax

  Run PyTorch in JAX. 🤝

  Language:Python233677

• neuraloperator/Geo-FNO

  Geometry-Aware Fourier Neural Operator (Geo-FNO)

  Language:Jupyter Notebook23241349

• jax-ml/scaling-book

  Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs

  Language:HTML2287434

• ucl-bug/jwave

  A JAX-based research framework for differentiable and parallelizable acoustic simulations, on CPU, GPUs and TPUs

  Language:Python16385921

• pnkraemer/matfree

  Matrix-free linear algebra in JAX.

  Language:Python1161743

• BirkhoffG/jax-dataloader

  Pytorch-like dataloaders for JAX.

  Language:Jupyter Notebook783203

• rdyro/torch2jax

  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.

  Language:Python492211

• tum-pbs/ConFIG

  [ICLR2025 Spotlight] Official implementation of Conflict-Free Inverse Gradients Method

  Language:Python491122

• cpraveen/na

  Codes for Numerical Analysis course

  Language:Jupyter Notebook362020

• evanatyourservice/psgd_jax

  Implementation of PSGD optimizer in JAX

  Language:Python30402

• Ceyron/exponax

  Efficient Differentiable n-d PDE solvers in JAX.

  Language:Jupyter Notebook283381

• nickmcgreivy/WeakBaselinesMLPDE

  Language:Python25102

• Ceyron/pdequinox

  Neural Emulator Architectures in JAX.

  Language:Python16420

• jejjohnson/jaxsw

  Simple differentiable approximate ocean models built with JAX.

  Language:Jupyter Notebook134411

• Ceyron/trainax

  Training methodologies for autoregressive neural operators/emulators in JAX.

  Language:Python7310

• Siftman/semantic-fashion-search

  This project utilizes clip ai model for semantic search - ( Text <-> Image) & ( Image <-> Image )

  Language:Python60

• Siftman/access_plugin

  A plugin enabling secure access and integration with an online marketplace, supporting account management, product listing, and data sync through a user-friendly interface.

  Language:PHP4100

• tum-pbs/PRDP

  PRDP paper

  Language:Jupyter Notebook21

• PredictiveIntelligenceLab/LieAD

  1